Update app.py

app.py

@@ -12,7 +12,8 @@ model = AutoModel.from_pretrained(
     MODEL_NAME,
     trust_remote_code=True,
     torch_dtype=torch.float32,
-    low_cpu_mem_usage=True
+    low_cpu_mem_usage=True,
+    attn_implementation="eager"
 )
 
 model = model.eval()