Spaces:
Runtime error
Runtime error
Commit ·
47f35d3
1
Parent(s): 5516bb8
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def predict_bounding_boxes(imageurl:str):
|
|
| 29 |
inputs = caption_processor(raw_image, return_tensors="pt")
|
| 30 |
out = caption_model.generate(**inputs)
|
| 31 |
label = caption_processor.decode(out[0], skip_special_tokens=True)
|
| 32 |
-
return {"label": label, "detections": detections}
|
| 33 |
|
| 34 |
except Exception as e:
|
| 35 |
|
|
|
|
| 29 |
inputs = caption_processor(raw_image, return_tensors="pt")
|
| 30 |
out = caption_model.generate(**inputs)
|
| 31 |
label = caption_processor.decode(out[0], skip_special_tokens=True)
|
| 32 |
+
return {"image label": label, "detections": detections}
|
| 33 |
|
| 34 |
except Exception as e:
|
| 35 |
|