Update app.py

app.py

@@ -3,17 +3,18 @@ from transformers import AutoModelForCausalLM, AutoProcessor, GenerationConfig
 from PIL import Image
 import torch
 import spaces
+allenai
 
 # Load the processor and model
 processor = AutoProcessor.from_pretrained(
-    'allenai/Molmo-7B-D-0924',
+    'allenai/MolmoE-1B-0924',
     trust_remote_code=True,
     torch_dtype='auto',
     device_map='auto'
 )
 
 model = AutoModelForCausalLM.from_pretrained(
-    'allenai/Molmo-7B-D-0924',
+    'allenai/MolmoE-1B-0924',
     trust_remote_code=True,
     torch_dtype='auto',
     device_map='auto'