Spaces:
Runtime error
Runtime error
comma
Browse files
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 =
|
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]
|