Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -127,8 +127,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 127 |
|
| 128 |
#def load_models():
|
| 129 |
# with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
|
| 130 |
-
florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-
|
| 131 |
-
florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-
|
| 132 |
# return florence_model, florence_processor
|
| 133 |
|
| 134 |
#florence_model, florence_processor = load_models()
|
|
|
|
| 127 |
|
| 128 |
#def load_models():
|
| 129 |
# with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
|
| 130 |
+
florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-Large', trust_remote_code=True).to(device).eval()
|
| 131 |
+
florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-Large', trust_remote_code=True)
|
| 132 |
# return florence_model, florence_processor
|
| 133 |
|
| 134 |
#florence_model, florence_processor = load_models()
|