back to gradio 4

README.md

@@ -4,7 +4,7 @@ emoji: 🌈
 colorFrom: red
 colorTo: blue
 sdk: gradio
-sdk_version: 5.0.0b3
+sdk_version: 4.44.0
 app_file: app.py
 pinned: true
 license: agpl-3.0

flagging.py

@@ -3,7 +3,7 @@ import json
 import uuid
 from collections import OrderedDict
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Sequence
+from typing import Any, Sequence
 
 import filelock
 import huggingface_hub

model_yolov5.py

@@ -1,4 +1,4 @@
-# import spaces
+import spaces
 import numpy as np
 import yolov5
 from yolov5.utils.plots import Annotator, colors
@@ -12,7 +12,7 @@ def load_model(model_path, img_size=640):
     return model
 
 
-# @spaces.GPU
+@spaces.GPU
 def inference(model, image):
     """Run inference on image and return annotated image."""
     results = model(image, size=model.img_size)