Update app.py

app.py

@@ -46,13 +46,15 @@ with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
     except Exception as e:
         print(f"Failed to load model: {str(e)}")
 
-    # Load the processor
-    processor = AutoProcessor.from_pretrained(
-        model_dir, 
-        trust_remote_code=True,
-        revision='refs/pr/6'
-    )
-    print("Processor loaded successfully!")
+    # Load the processor without specifying a revision
+    try:
+        processor = AutoProcessor.from_pretrained(
+            model_dir, 
+            trust_remote_code=True
+        )
+        print("Processor loaded successfully!")
+    except Exception as e:
+        print(f"Failed to load processor: {str(e)}")
 TITLE = "# [Florence-2-DocVQA Demo](https://huggingface.co/HuggingFaceM4/Florence-2-DocVQA)"
 DESCRIPTION = "The demo for Florence-2 fine-tuned on DocVQA dataset. You can find the notebook [here](https://colab.research.google.com/drive/1hKDrJ5AH_o7I95PtZ9__VlCTNAo1Gjpf?usp=sharing). Read more about Florence-2 fine-tuning [here](finetune-florence2)."