Update app.py

app.py

@@ -1,20 +1,18 @@
-import gradio as gr
 from transformers import pipeline
+from huggingface_hub import login
 
-# Load the model (change the model name if needed)
-model_name = "google/gemma-2b-it"
-generator = pipeline("text-generation", model=model_name)
+# OPTIONAL: Authenticate with your Hugging Face API token (if needed)
+# login("your_huggingface_api_token_here")  
+
+# Load the Mistral-7B-Instruct model
+model_name = "mistralai/Mistral-7B-Instruct"
+generator = pipeline("text-generation", model=model_name, device="cpu")  # Change "cpu" to "cuda" if using a GPU
 
 def generate_response(prompt):
     response = generator(prompt, max_length=200)
     return response[0]["generated_text"]
 
-iface = gr.Interface(
-    fn=generate_response,
-    inputs="text",
-    outputs="text",
-    title="Waste Reuse AI",
-    description="This AI suggests reuse ideas for waste items."
-)
-
-iface.launch()
+# Example Usage
+if __name__ == "__main__":
+    user_prompt = "How can I reuse a plastic bottle?"
+    print("AI Suggestion:", generate_response(user_prompt))