Update app.py

app.py

@@ -1,28 +1,41 @@
+# ============================================
+# 💜 NasFit Vision AI — Dark Neon Edition
+# ============================================
+
+# 🚧 Asegúrate de tener este contenido en tu requirements.txt:
+# transformers>=4.43.0
+# torch
+# accelerate
+# gradio
+# Pillow
+
 import gradio as gr
 import torch
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoModelForVision2Seq
+from transformers import LlavaOneVisionProcessor  # 👈 clase correcta
 from PIL import Image
 import re
-from io import BytesIO
 
 # ============================================
-# 🔮 CONFIGURACIÓN DEL MODELO
+# 🔮 CARGA DEL MODELO LOCAL
 # ============================================
 
 MODEL_ID = "lmms-lab/llava-onevision-1.5-8b-instruct"
 
 print("⏳ Cargando modelo local con trust_remote_code=True...")
-processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
+
+processor = LlavaOneVisionProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
 model = AutoModelForVision2Seq.from_pretrained(
     MODEL_ID,
     trust_remote_code=True,
     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     device_map="auto"
 )
-print("✅ Modelo cargado correctamente en modo local.")
+
+print("✅ Modelo LLaVA-OneVision cargado correctamente.")
 
 # ============================================
-# 🧠 FUNCIONES DE ANÁLISIS
+# 🧠 FUNCIONES DE ANÁLISIS NUTRICIONAL
 # ============================================
 
 def extract_macros(text):
@@ -77,7 +90,7 @@ def analyze_food(image, text_prompt="Describe esta comida y estima sus calorías
         return f"<div class='card error'>⚠️ Error: {e}</div>"
 
 # ============================================
-# 💅 INTERFAZ DE USUARIO (Glass Neon)
+# 💅 INTERFAZ DE USUARIO (Purple Glassmorphism)
 # ============================================
 
 def build_interface():