Upload 3 files

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+import cv2
+import numpy as np
+from ultralytics import YOLO
+
+# Load model
+model = YOLO("railway_track_segment.pt")
+
+def segment_track(image):
+    """Segment railway track from image"""
+    # Predict on CPU (no GPU available)
+    results = model.predict(image, conf=0.25, imgsz=640, device="cpu")
+    
+    # Get segmented image
+    segmented = results[0].plot(boxes=False, conf=True, line_width=2)
+    segmented = cv2.cvtColor(segmented, cv2.COLOR_BGR2RGB)
+    
+    return segmented
+
+# Create interface
+demo = gr.Interface(
+    fn=segment_track,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="numpy"),
+    title="Railway Track Segmentation",
+    description="Upload a railway track image to segment it"
+)
+
+# Launch
+demo.launch(share=True)

railway_track_segment (1).pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82d3af89be96e8193c564ba39e69634160de99ae642c4f928ff9f4332d667db5
+size 6012701

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+ultralytics