Update app.py

app.py

@@ -2,36 +2,31 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# 🔹 Clave API de Hugging Face (se guarda como variable secreta en Hugging Face)
-API_KEY = os.getenv("HF_API_KEY")  # Debes guardarlo en "Secrets" de tu Space
+# 🔹 API Key de Hugging Face (debe estar en las Variables del Space)
+API_KEY = os.getenv("HF_API_KEY")
 
-# 🔹 Cliente de inferencia usando Hugging Face Hyperbolic
+# 🔹 Cliente de Hugging Face Inference
 client = InferenceClient(
-    provider="hf-inference",
-    api_key=API_KEY  # Asegúrate de que la variable se llama API_KEY
+    model="flax-community/spanish-t5-small",
+    api_key=API_KEY  # Usa la API Key de Hugging Face
 )
 
-# 🔒 Clave secreta de acceso para los usuarios
-SECRET_KEY = "MI_CLAVE_SUPER_SECRETA"  # Cambia esto por una clave segura
+# 🔒 Clave secreta de acceso
+SECRET_KEY = "MI_CLAVE_SUPER_SECRETA"
 
 # 🔹 Función para reformular el texto
 def reformular_texto(frase, clave):
     if clave != SECRET_KEY:
         return "🔴 Acceso denegado."
 
-messages = [
-    {"role": "user", "content": f"Convierte esta frase en español a una gramática correcta sin cambiar su significado: {frase}"}
-]
+    # 🔥 Prompt mejorado
+    prompt = f"Convierte esta frase en español a una gramática correcta sin cambiar su significado: {frase}"
 
     try:
-        completion = client.chat.completions.create(
-            model="deepseek-ai/DeepSeek-R1",
-            messages=messages,
-            max_tokens=50,
-        )
-        return completion.choices[0].message.content  # Devuelve el texto reformulado
+        completion = client.text_generation(prompt, max_new_tokens=50)
+        return completion
     except Exception as e:
-        return f"⚠️ Error en la inferencia: {str(e)}"
+        return f"⚠ Error en la inferencia: {str(e)}"
 
 # 🔹 Interfaz con Gradio
 iface = gr.Interface(