Update app.py

app.py

@@ -5,17 +5,20 @@ import os
 token = os.environ.get("HF_TOKEN")
 
 client = InferenceClient(
-    model="mistralai/Mistral-7B-Instruct-v0.3",
+    model="meta-llama/Llama-3.2-3B-Instruct",
     token=token
 )
 
 def generate(prompt):
-    response = client.chat_completion(
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=512,
-        temperature=0.7
-    )
-    return response.choices[0].message.content
+    try:
+        response = client.chat_completion(
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=512,
+            temperature=0.7
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 gr.Interface(
     fn=generate,