Upload app.py

app.py

@@ -4,17 +4,15 @@ from huggingface_hub import InferenceClient
 api_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
 client = InferenceClient(
-    model="google/flan-t5-large",
+    model="mistralai/Mistral-7B-Instruct-v0.2",  # Example chat model
     token=api_token
 )
 
 prompt = "Hello world"
 
-# You can also specify additional parameters
-response = client.text_generation(
-    prompt=prompt,
-    max_new_tokens=50,
-    temperature=0.7
+# For chat models
+response = client.chat_completion(
+    messages=[{"role": "user", "content": prompt}]
 )
 
-print(response)
+print(response.choices[0].message.content)