added all necessary files for application

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+.pth filter=lfs diff=lfs merge=lfs -text
+.jpg filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,56 @@
+import gradio as gr
+import supervision as sv
+from rfdetr import RFDETRBase
+
+models = {
+    "GC10-DET": "src/models/gc10-det.pth",
+    "NEU-DET": "src/models/neu-det.pth"
+}
+
+def predict(model_name, img, conf, iou):
+    model_path = models[model_name]
+    model = RFDETRBase(pretrain_weights=model_path)
+    model.optimize_for_inference()
+
+    results = model.predict(img, threshold=conf)
+    results = results.with_nmm(threshold=iou)
+    
+    labels = [f"defect {confidence:.2f}" for confidence in results.confidence]
+    
+    annotated_image = img.copy()
+    annotated_image = sv.BoxAnnotator().annotate(annotated_image, results)
+    annotated_image = sv.LabelAnnotator().annotate(annotated_image, results, labels)
+
+    return annotated_image
+
+base_conf, base_iou = 0.25, 0.5
+title = "Detection with RF-DETR trained on NEU-DET and GC10-DET"
+des = """
+        **Instructions:**
+        1. Choose a model:  
+        \- **GC10-DET** - for images from the GC10 dataset (10 defect types, steel surface images)  
+        \- **NEU-DET** - for images from the NEU dataset (6 defect types, steel strip images)
+        2. Upload Image.
+        3. Adjust thresholds
+        4. Click the **Submit** button. Wait while predictions are generated.
+        """
+interface = gr.Interface(
+    fn=predict,
+    inputs=[
+        gr.Dropdown(list(models.keys()), label="Select Model"),
+        'image', 
+        gr.Slider(maximum=1, minimum=0, value=base_conf, label="Confidence Threshold"), 
+        gr.Slider(maximum=1, minimum=0, value=base_iou, label="NMM IoU Threshold")
+    ],
+    outputs=["image"],
+    title=title,
+    description=des,
+    examples=[
+        ["GC10-DET", "example1.jpg", base_conf, base_iou],
+        ["GC10-DET", "example2.jpg", base_conf, base_iou],
+        ["NEU-DET", "example3.jpg", base_conf, base_iou]
+    ]
+)
+
+interface.launch()
+

gc10-det.pth

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

neu-det.pth

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

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+git+https://github.com/roboflow/rf-detr.git