Update app.py

app.py

@@ -1,35 +0,0 @@
-import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-import gradio as gr
-
-model_id = "teknium/OpenHermes-2.5-Mistral-7B"
-
-# Carga el modelo y tokenizer
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(
-    model_id,
-    device_map="auto",
-    torch_dtype=torch.float16,
-)
-
-pipe = pipeline(
-    "text-generation",
-    model=model,
-    tokenizer=tokenizer,
-    max_new_tokens=256,
-    do_sample=True,
-    temperature=0.7,
-)
-
-# Interfaz Gradio
-def generate_text(prompt):
-    output = pipe(prompt)[0]["generated_text"]
-    return output
-
-gr.Interface(
-    fn=generate_text,
-    inputs=gr.Textbox(label="prompt"),
-    outputs=gr.Textbox(label="output"),
-    title="Amside AI",
-    theme="soft",
-).launch()