Working on getting the outputs
Browse files
app.py
CHANGED
|
@@ -77,6 +77,10 @@ def segment_images(model_name,url_input,image_input,threshold):
|
|
| 77 |
model = MaskFormerForInstanceSegmentation.from_pretrained(model_name)
|
| 78 |
|
| 79 |
inputs = processor(images=image, return_tensors="pt")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 80 |
pass
|
| 81 |
else:
|
| 82 |
raise NameError("Model is not implemented")
|
|
|
|
| 77 |
model = MaskFormerForInstanceSegmentation.from_pretrained(model_name)
|
| 78 |
|
| 79 |
inputs = processor(images=image, return_tensors="pt")
|
| 80 |
+
|
| 81 |
+
outputs = model(**inputs)
|
| 82 |
+
print(type(outputs))
|
| 83 |
+
print(outputs)
|
| 84 |
pass
|
| 85 |
else:
|
| 86 |
raise NameError("Model is not implemented")
|