Update app.py

app.py

@@ -73,9 +73,12 @@ def response_gen(user_review):
     # Format the prompt with the user's review
     prompt = emotion_prompts.get(emotion_label, "Neutral").format(review=user_review)
     
-    # Load a pre-trained text generation model (replace 'meta-llama/Llama-3.2-1B' with an available model)
-    tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
-    model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")
+    # Load model directly
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+
+    tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-0.5B")
+    model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen1.5-0.5B")
+    
     inputs = tokenizer(prompt, return_tensors="pt")  # Tokenize the prompt
     outputs = model.generate(**inputs, max_new_tokens=100)  # Generate a response