Spaces:

ArchiMathur
/

fireGradio

Runtime error

ArchiMathur commited on Oct 27, 2024

Commit

3a8bc47

verified ·

1 Parent(s): 258d3a6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import cv2
 import numpy as np
 import gradio as gr
 import torch
 # Load the YOLO model (adjust the path to your model weights and config)
 model = torch.hub.load('best.pt')  # Change to your model path

 import numpy as np
 import gradio as gr
 import torch
+from ultralytics import YOLO
 # Load the YOLO model (adjust the path to your model weights and config)
 model = torch.hub.load('best.pt')  # Change to your model path