Update app.py

app.py

@@ -18,6 +18,7 @@ from PIL import Image
 from transformers import DALLEncoder, DALLDecoder
 import uuid
 from tqdm import tqdm
+from transformers import eBart
 
 logging.basicConfig(level=logging.INFO)
 
@@ -36,15 +37,7 @@ print("Dispositivo:", device)
 modelos = {
     "gpt2-medium": VLLM(model="gpt2-medium"),
     "qwen2.5-0.5b": VLLM(model="Qwen/Qwen2.5-0.5B-Instruct"),
-    "llamaxd": VLLM(model="Hjgugugjhuhjggg/llama-3.2-1B-spinquant-hf"),
-    "t5-base": VLLM(model="t5-base"),
-    "bert-base-uncased": VLLM(model="bert-base-uncased"),
-    "musicgen-small": VLLM(model="musicgen-small"),
-    "dall-e-mini": VLLM(model="dall-e-mini"),
-    "xlnet-base-uncased": VLLM(model="xlnet-base-uncased"),
-    "distilbert-base-uncased": VLLM(model="distilbert-base-uncased"),
-    "albert-base-v2": VLLM(model="albert-base-v2"),
-    "roberta-base": VLLM(model="roberta-base"),
+    "llamaxd": VLLM(model="Hjgugugjhuhjggg/llama-3.2-1B-spinquant-hf")
 }
 
 print("Cargando modelos...")
@@ -76,8 +69,8 @@ vectorizer = TfidfVectorizer()
 
 print("Cargando modelo de vectorizador TF-IDF...")
 
-dalle_encoder = DALLEncoder(model_id="dall-e-mini")
-dalle_decoder = DALLDecoder(model_id="dall-e-mini")
+dalle_encoder = eBart.from_pretrained(model_id="dalle-mini/dalle-mini")
+dalle_decoder = eBart.from_pretrained(model_id="dalle-mini/dalle-mini")
 
 print("Cargando modelo DALL-E...")