Update app.py
Browse files
app.py
CHANGED
@@ -15,4 +15,10 @@ with torch.no_grad():
|
|
15 |
|
16 |
# model predicts one of the 1000 ImageNet classes
|
17 |
predicted_label = logits.argmax(-1).item()
|
18 |
-
print(model.config.id2label[predicted_label])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
# model predicts one of the 1000 ImageNet classes
|
17 |
predicted_label = logits.argmax(-1).item()
|
18 |
+
print(model.config.id2label[predicted_label])
|
19 |
+
|
20 |
+
import gradio as gr
|
21 |
+
def segment(image):
|
22 |
+
pass # Implement your image segmentation model here...
|
23 |
+
|
24 |
+
gr.Interface(fn=segment, inputs="image", outputs="image").launch()
|