Update app.py

app.py

@@ -23,18 +23,7 @@ import gradio as gr
 feature_extractor = SegformerFeatureExtractor.from_pretrained('nvidia/segformer-b0-finetuned-ade-512-512')
 segformer_model = SegformerForSemanticSegmentation.from_pretrained('nvidia/segformer-b0-finetuned-ade-512-512')
 
-# Inference function for segmentation
-def segment_image(image):
-    inputs = feature_extractor(images=image, return_tensors="pt")
-    outputs = segformer_model(**inputs)
-    segmentation = outputs.logits.argmax(dim=1).squeeze().cpu().numpy()
-    return segmentation
-
-# Gradio interface at the end of the script
-iface = gr.Interface(fn=segment_image, inputs=gr.Image(type="pil"), outputs="image")
-
-# Specify a custom port if needed to avoid conflicts (optional)
-iface.launch(server_port=7861)  # Change port if 7860 is occupied
+
 
 
 # Function to extract zip files