Florence-2-base-Castollux-v0.6

Running

xzuyn commited on Feb 7

Commit

240d96f

verified ·

1 Parent(s): 74d81c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
-TITLE = "# [PJMixers-Images/Florence-2-base-Castollux-v0.5](https://huggingface.co/PJMixers-Images/Florence-2-base-Castollux-v0.5)"
 def process_image(image):

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
+TITLE = f"# [{model_name}](https://huggingface.co/{model_name})"
 def process_image(image):