Upload app.py

app.py

@@ -4,15 +4,13 @@ from huggingface_hub import InferenceClient
 api_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
 client = InferenceClient(
-    model="mistralai/Mistral-7B-Instruct-v0.2",  # Example chat model
+    model="google/flan-t5-large",
     token=api_token
 )
 
 prompt = "Hello world"
 
-# For chat models
-response = client.chat_completion(
-    messages=[{"role": "user", "content": prompt}]
-)
+# Use text generation method instead
+response = client.text_generation(prompt)
 
-print(response.choices[0].message.content)
+print(response)