jadechoghari commited on
Commit
c30e671
·
verified ·
1 Parent(s): 8801c9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -10,7 +10,7 @@ detector = pipeline(model=checkpoint, task="zero-shot-object-detection")
10
  sam_model = SamModel.from_pretrained("jadechoghari/robustsam-vit-base")
11
  sam_processor = SamProcessor.from_pretrained("jadechoghari/robustsam-vit-base")
12
 
13
- @spaces.GPU
14
  def query(image, texts, threshold):
15
  texts = texts.split(",")
16
 
@@ -69,6 +69,10 @@ demo = gr.Interface(
69
  inputs=[gr.Image(type="pil", label="Image Input"), gr.Textbox(label = "Candidate Labels"), gr.Slider(0, 1, value=0.05, label="Confidence Threshold")],
70
  outputs="annotatedimage",
71
  title="OWL 🤝 SAM",
72
- description=description
 
 
 
 
73
  )
74
  demo.launch(debug=True)
 
10
  sam_model = SamModel.from_pretrained("jadechoghari/robustsam-vit-base")
11
  sam_processor = SamProcessor.from_pretrained("jadechoghari/robustsam-vit-base")
12
 
13
+
14
  def query(image, texts, threshold):
15
  texts = texts.split(",")
16
 
 
69
  inputs=[gr.Image(type="pil", label="Image Input"), gr.Textbox(label = "Candidate Labels"), gr.Slider(0, 1, value=0.05, label="Confidence Threshold")],
70
  outputs="annotatedimage",
71
  title="OWL 🤝 SAM",
72
+ description=description,
73
+ examples=[
74
+ ["./cats.png", "cat", 0.1],
75
+ ],
76
+ cache_examples=True
77
  )
78
  demo.launch(debug=True)