Update app.py

app.py

@@ -6,11 +6,11 @@ device = 0 if torch.cuda.is_available() else -1
 
 def generate_response(user_input, history, temperature=0.5, do_sample=True):
     pipe = pipeline("text-generation", model="frameai/ChatFrame-Instruct-Persian-Small", device=device)
-    pipe.set_params(temperature=temperature, do_sample=do_sample)
+
     messages = [
         {"role": "user", "content": user_input},
     ]
-    response = pipe(messages, max_length=1024)
+    response = pipe(messages, max_length=1024, temperature=temperature, do_sample=do_sample)
     return response[0]['generated_text'][1]["content"]
 
 iface = gr.ChatInterface(