Spaces:

piky
/

zero-shot-object-detection

Paused

piky commited on Apr 24

Commit

4dceaa5

verified ·

1 Parent(s): 998fc2d

Update app.py

Guess two capsules

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)
-image_url = "https://pub-44a464d4394a43d6834ffdc08038cb38.r2.dev/static/img/guess_a_capsule.jpeg"
 headers = {
     "User-Agent": "Mozilla/5.0"

 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)
+image_url = "https://pub-44a464d4394a43d6834ffdc08038cb38.r2.dev/static/img/guess_two_capsules.jpg"
 headers = {
     "User-Agent": "Mozilla/5.0"