Update app.py

app.py

@@ -35,13 +35,13 @@ def save_state_to_disk(dataframe_rows):
         print(f"⚠️ Critical storage write failure: {e}")
 
 # =====================================================================
-# 🛰️ GLOBAL VISION-LANGUAGE INITIALIZATION
+# 🛰️ GLOBAL VISION-LANGUAGE INITIALIZATION (Class Mapping Corrected)
 # =====================================================================
 print("🤖 Initializing Moondream VLM weights into memory pool...")
 model_id = "vikhyatk/moondream2"
 
-# Keep the model load global. On ZeroGPU, this registers it in container RAM.
-model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
+# Using AutoModel instead of AutoModelForCausalLM bypasses the tied weights checking loop
+model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 print("Model layers successfully loaded.")