Update app.py

app.py

@@ -121,13 +121,13 @@ DEFAULT_NEGATIVE_SUFFIX = "Nsfw oversaturated crappy_art low_quality blurry bad_
 # Initialize Florence model
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-def load_models():
-    with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
-        florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()
-        florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
-    return florence_model, florence_processor
+#def load_models():
+    #with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
+florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()
+florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
+    #return florence_model, florence_processor
 
-florence_model, florence_processor = load_models()
+#florence_model, florence_processor = load_models()
 
 # Prompt Enhancer
 enhancer_medium = pipeline("summarization", model="gokaygokay/Lamini-Prompt-Enchance", device=device)