kinsung commited on
Commit
628652c
·
1 Parent(s): 7374afb
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ o2 = gr.outputs.Textbox(label="Similarity score")
18
  def extract_image(image, text, prob, custom_width, custom_height):
19
  inputs = feature_extractor(images=image, return_tensors="pt")
20
  outputs = dmodel(**inputs)
21
- pout = None
22
 
23
  target_sizes = torch.tensor([image.size[::-1]])
24
  results = feature_extractor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]
 
18
  def extract_image(image, text, prob, custom_width, custom_height):
19
  inputs = feature_extractor(images=image, return_tensors="pt")
20
  outputs = dmodel(**inputs)
21
+ pout = ""
22
 
23
  target_sizes = torch.tensor([image.size[::-1]])
24
  results = feature_extractor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]