Update app.py

app.py

@@ -105,7 +105,7 @@ def check_and_load_model(model_name):
         return llm
 
 # Interface Gradio
-def gradio_interface(model_name, input, max_tokens, temperature, history, stop_button=None):
+def gradio_interface(model_name, input, max_tokens, temperature, history):
     global stop_flag
     stop_flag = False
     response = None  # Initialisez la variable response ici
@@ -120,7 +120,8 @@ def gradio_interface(model_name, input, max_tokens, temperature, history, stop_b
     thread.join()
 
     if stop_flag:
-        return "Processing stopped by the user.", history + [[input, "Processing stopped by the user."]]
+        history.append((input, "Processing stopped by the user."))
+        return "Processing stopped by the user.", history
     else:
         history.append((input, response))
         return response, history  # Maintenant, response est accessible ici
@@ -134,11 +135,11 @@ demo = gr.Interface(
         gr.Textbox(label="Input"),
         gr.Slider(minimum=1, maximum=2048, step=1, label="Max Tokens", value=256),
         gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label="Temperature", value=0.7),
-        gr.Button(value="Stop", variant="stop")
+        gr.State(value=[])  # Ajout de l'historique
     ],
     outputs=[
         gr.Textbox(label="Output"),
-        gr.State(value=[])  # Ajout de l'historique
+        gr.State()  # Ajout de l'historique
     ],
     title="Ollama Demo",
     description="""
@@ -147,7 +148,6 @@ demo = gr.Interface(
     Cette Démo permet aux utilisateurs de tester tous les modèles Ollama dont la taille est inférieure à 10 milliards de paramètres directement depuis cette interface.
 
     L'Application tourne sur une machine Hugging Face Free Space : 2 CPU - 16Gb RAM
-    
     Soyez patient...
     """
 )