Update app.py

app.py

@@ -1,20 +1,13 @@
 import torch
+import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from fastapi import FastAPI
-from pydantic import BaseModel
 
 # Charger le modèle fine-tuné
 MODEL_NAME = "fatmata/psybot"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
 
-print("✅ Modèle et tokenizer chargés avec succès !")  # Vérification du chargement
-
-# Définir l'API avec FastAPI
-app = FastAPI()
-
-class UserInput(BaseModel):
-    text: str
+print("✅ Modèle chargé avec succès :", model.config)  # Debugging
 
 def generate_response(user_input):
     """ Génère une réponse du chatbot PsyBot """
@@ -44,7 +37,7 @@ def generate_response(user_input):
     
     return response
 
-@app.post("/generate/")
-def generate(user_input: UserInput):
-    response = generate_response(user_input.text)
-    return {"response": response}
+# Interface Gradio avec le bon modèle
+iface = gr.Interface(fn=generate_response, inputs="text", outputs="text")
+
+iface.launch(server_name="0.0.0.0", server_port=7860)