Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Charger le modèle et le tokenizer
@@ -6,10 +6,15 @@ model_name = "Rimyy/MISTRAL-finetuneGSMdata1exp"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+# Fonction de prédiction
 def predict(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(inputs.input_ids, max_length=256)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-interface = gr.Interface(fn=predict, inputs="text", outputs="text")
-interface.launch()
+# Interface utilisateur Streamlit
+st.title("Modèle de génération de texte")
+prompt = st.text_area("Entrez votre texte:")
+if st.button("Générer"):
+    result = predict(prompt)
+    st.text_area("Résultat", value=result, height=200)