Update modules/masking_module.py

modules/masking_module.py

@@ -15,7 +15,7 @@ device = torch.device('cuda')
 model_id = 'microsoft/Florence-2-large'
 
 models_dict = {
-    'Florence_model':AutoModelForCausalLM.from_pretrained(model_id', trust_remote_code=True).to("cuda").eval(),
+    'Florence_model':AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cuda").eval(),
     'Florence_processor':AutoProcessor.from_pretrained(model_id, trust_remote_code=True),
 }