Update app.py

app.py

@@ -1,22 +1,19 @@
 import os
 import gradio as gr
-from huggingface_hub import InferenceApi
+from huggingface_hub import InferenceClient
 
 model_id = 'dicta-il/dictalm-7b-instruct'
 
 # יצירת API של Hugging Face
 api_key = os.getenv('HUGGINGFACE_API_KEY', '')
-generator = InferenceApi(repo_id=model_id, token=api_key)
+generator = InferenceClient(model=model_id, token=api_key)
 
 # פונקציית יצירת הטקסט
 def chat_with_model(history):
     prompt = history[-1]["content"]
     try:
-        response = generator(prompt)
-        if isinstance(response, dict) and "generated_text" in response:
-            result = response["generated_text"]
-        else:
-            result = response
+        response = generator.text_generation(prompt, do_sample=True, min_length=20, max_length=64, top_k=40, top_p=0.92, temperature=0.9)
+        result = response.get("generated_text", "Error: No generated text found")
     except Exception as e:
         result = f"Error: {str(e)}"
     return history + [{"role": "bot", "content": result}]