Update model_api.py

model_api.py

@@ -6,20 +6,17 @@ def query_model(prompt):
         HF_TOKEN = os.getenv("HF_TOKEN")
 
         client = InferenceClient(
-           model="HuggingFaceH4/zephyr-7b-beta",
-           token=HF_TOKEN
-)
+            model="google/flan-t5-large",
+            token=HF_TOKEN
+        )
 
-        response = client.chat_completion(
-            messages=[
-                {"role": "system", "content": "You are a certified professional fitness trainer."},
-                {"role": "user", "content": prompt}
-            ],
-            max_tokens=1500,
+        response = client.text_generation(
+            prompt,
+            max_new_tokens=1200,
             temperature=0.7
         )
 
-        return response.choices[0].message.content
+        return response
 
     except Exception as e:
         return f"Error: {str(e)}"