Added yolov7 model

app.py

@@ -7,9 +7,9 @@ import os
 
 os.system('pip install git+https://github.com/fcakyon/ultralyticsplus.git')
 
-model_id_list = ['deprem-ml/Binafarktespit-yolo5x-v1-xview', 'SerdarHelli/deprem_satellite_labeled_yolov8']
+model_id_list = ['deprem-ml/Binafarktespit-yolo5x-v1-xview', 'SerdarHelli/deprem_satellite_labeled_yolov8', 'kadirnar/yolov7-v0.1']
 current_device = "cuda" if torch.cuda.is_available() else "cpu"
-model_types = ["YOLOv5", "YOLOv5 + SAHI", "YOLOv8"]
+model_types = ["YOLOv5", "YOLOv5 + SAHI", "YOLOv8", "YOLOv7"]
 
 def sahi_yolov5_inference(
     image,
@@ -92,6 +92,13 @@ def sahi_yolov5_inference(
         result = model.predict(image, imgsz=image_size)[0]
         render = render_result(model=model, image=image, result=result, rect_th=rect_th, text_th=text_th)
         return render
+    
+    elif model_type == "YOLOv7":
+        import yolov7
+        
+        model = yolov7.load(model_id, device="cuda:0", hf_model=True, trace=False)
+        results = model([image], size=image_size)
+        return results.render()[0]
 
 inputs = [
     gr.Image(type="pil", label="Original Image"),

requirements.txt

@@ -1,4 +1,5 @@
 torch==1.10.2
 torchvision==0.11.3
 yolov5==7.0.8
-sahi==0.11.11
+sahi==0.11.11
+yolov7detect==1.0.1