Update app.py

app.py

@@ -21,7 +21,7 @@ model = AutoModel.from_pretrained(
     # load_in_8bit=True,
     low_cpu_mem_usage=True,
     use_flash_attn=True,
-    trust_remote_code=True).eval()
+    trust_remote_code=True).eval().to("cpu")
 tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True, use_fast=False)
 
 def preprocessing(image):