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")
|