Update model_api.py

model_api.py

@@ -13,13 +13,16 @@ def query_model(prompt):
             token=HF_TOKEN
         )
 
-        response = client.text_generation(
-            prompt,
-            max_new_tokens=300,
+        response = client.chat_completion(
+            messages=[
+                {"role": "system", "content": "You are a certified professional fitness trainer."},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=400,
             temperature=0.7
         )
 
-        return response
+        return response.choices[0].message.content
 
     except Exception as e:
         return f"Error: {str(e)}"