xzuyn commited on
Commit
27c0fae
·
verified ·
1 Parent(s): 2fb1c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoProcessor, Florence2ForConditionalGeneration
11
  # Load model and processor, enabling trust_remote_code if needed
12
  model_name = "PJMixers-Images/Florence-2-base-Castollux-v0.5"
13
  model = Florence2ForConditionalGeneration.from_pretrained(model_name).eval()
14
- processor = AutoProcessor.from_pretrained(model_nam)
15
 
16
  # Set device (GPU if available)
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
11
  # Load model and processor, enabling trust_remote_code if needed
12
  model_name = "PJMixers-Images/Florence-2-base-Castollux-v0.5"
13
  model = Florence2ForConditionalGeneration.from_pretrained(model_name).eval()
14
+ processor = AutoProcessor.from_pretrained(model_name)
15
 
16
  # Set device (GPU if available)
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")