Update app.py

app.py

@@ -1,43 +1,16 @@
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM, TextGenerationPipeline
-from diffusers import StableDiffusionPipeline
-from PIL import Image
-import io
 
-# Modelo de texto en CPU
-text_model = "tiiuae/falcon-rw-1b"
-tokenizer = AutoTokenizer.from_pretrained(text_model)
-model = AutoModelForCausalLM.from_pretrained(text_model)
-text_pipeline = TextGenerationPipeline(model=model, tokenizer=tokenizer, device=-1)
+model_id = "mistralai/Mistral-7B-Instruct-v0.1"  # Requiere acceso
+tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16, use_auth_token=True)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Modelo de imagen en CPU
-image_pipe = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", 
-    torch_dtype=torch.float32
-).to("cpu")
+def chat(user_input):
+    prompt = f"""<s>[INST] {user_input.strip()} [/INST]"""
+    output = pipe(prompt, max_new_tokens=200, temperature=0.7, do_sample=True)[0]["generated_text"]
+    response = output.split("[/INST]")[-1].strip()
+    return response
 
-# Lógica para detectar si el prompt es de texto o imagen
-def chatbot(input_text):
-    if any(word in input_text.lower() for word in ["imagen", "dibuja", "pinta", "foto", "muestra"]):
-        image = image_pipe(input_text).images[0]
-        return None, image
-    else:
-        response = text_pipeline(input_text, max_new_tokens=150, do_sample=True)[0]['generated_text']
-        return response, None
-
-# Interfaz Gradio
-with gr.Blocks() as demo:
-    gr.Markdown("## Bot Generador de Texto e Imágenes (CPU)")
-
-    with gr.Row():
-        textbox = gr.Textbox(placeholder="Escribe algo... (ej: Dibuja una chica en la playa)")
-        send = gr.Button("Enviar")
-
-    with gr.Row():
-        text_output = gr.Textbox(label="Respuesta de texto")
-        image_output = gr.Image(label="Imagen generada")
-
-    send.click(fn=chatbot, inputs=textbox, outputs=[text_output, image_output])
-
-demo.launch()
+gr.Interface(fn=chat, inputs="text", outputs="text", title="MyBot - Texto").launch()