Fix app

app.py

@@ -1,7 +1,40 @@
 import gradio as gr
+from transformers import AutoTokenizer, pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+def initialize_model():
+    tokenizer = AutoTokenizer.from_pretrained("Nac31/Sacha-Mistral-0", trust_remote_code=True)
+    pipe = pipeline(
+        "text-generation",
+        model="Nac31/Sacha-Mistral-0",
+        tokenizer=tokenizer,
+        trust_remote_code=True
+    )
+    return pipe
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def generate_response(message, temperature=0.7, max_length=500):
+    try:
+        pipe = initialize_model()
+        response = pipe(
+            message,
+            max_length=max_length,
+            temperature=temperature,
+            do_sample=True
+        )[0]['generated_text']
+        return response
+    except Exception as e:
+        return f"Une erreur s'est produite : {str(e)}"
+
+# Interface Gradio
+demo = gr.Interface(
+    fn=generate_response,
+    inputs=[
+        gr.Textbox(label="Votre message", placeholder="Entrez votre message ici..."),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.7, label="Température")
+    ],
+    outputs=gr.Textbox(label="Réponse"),
+    title="Chat avec Sacha-Mistral",
+    description="Un assistant conversationnel en français basé sur le modèle Sacha-Mistral"
+)
+
+if __name__ == "__main__":
+    demo.launch()