Update app.py

app.py

@@ -6,11 +6,11 @@ import asyncio
 model_id = 'Aksh1t/mistral-7b-oig-unsloth-merged'
 
 # Initialize the Hugging Face inference client
-client = InferenceClient()
+client = InferenceClient(model=model_id)
 
 async def generate_text(prompt):
     # Use the Hugging Face client to generate text asynchronously
-    response = await client.text_generation(model=model_id, text=prompt)
+    response = await client.text_generation(prompt)
     return response['generated_text']
 
 # Create Gradio interface
@@ -25,3 +25,4 @@ iface = gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title="Hugging
 
 if __name__ == "__main__":
     iface.launch()
+