Update app.py

app.py

@@ -1,37 +1,16 @@
 import gradio as gr
-import requests
-import os
-import time
+from transformers import pipeline
 
-API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom-560m"
-headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
-
-def query(payload):
-    for _ in range(3):  # intenta 3 veces
-        response = requests.post(API_URL, headers=headers, json=payload)
-        result = response.json()
-
-        # Si el modelo está cargando, espera y reintenta
-        if isinstance(result, dict) and "error" in result:
-            time.sleep(5)
-            continue
-
-        return result
-
-    return {"error": "No response"}
+chatbot = pipeline("text-generation", model="distilgpt2")
 
 def respond(message, history):
     if not message:
         return "Pregúntame algo 😊"
 
-    output = query({
-        "inputs": "Responde en español y explica de forma sencilla: " + message
-    })
+    prompt = "Responde en español de forma clara: " + message
+    response = chatbot(prompt, max_length=100, do_sample=True)
 
-    if isinstance(output, list) and "generated_text" in output[0]:
-        return output[0]["generated_text"]
-    else:
-        return "La IA está ocupada 😭 intenta otra vez en unos segundos"
+    return response[0]["generated_text"]
 
 demo = gr.ChatInterface(fn=respond)