Update app.py

app.py

@@ -11,6 +11,15 @@ API_KEY = os.getenv("API_KEY") #or "YOUR_GEMINI_API_KEY"
 
 genai.configure(api_key=API_KEY)
 model = genai.GenerativeModel("gemini-1.5-pro")
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+pipe = pipeline("text-to-speech", model="saheedniyi/YarnGPT")
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("saheedniyi/YarnGPT")
+model = AutoModelForCausalLM.from_pretrained("saheedniyi/YarnGPT")
 
 def predict_emotion():
     return random.choice(["positive", "neutral", "negative"])