Aumkeshchy2003 commited on
Commit
0b7a6d2
·
verified ·
1 Parent(s): 84ae129

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -1,7 +1,3 @@
1
- !pip install -U torch torchvision cython
2
- !pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
3
- !pip install gradio
4
-
5
  import torch
6
  import torchvision
7
  from torchvision.transforms import functional as F
@@ -9,19 +5,13 @@ from PIL import Image
9
  import cv2
10
  import gradio as gr
11
 
12
- !git clone https://github.com/ultralytics/yolov5
13
- %cd yolov5
14
- !pip install -r requirements.txt
15
-
16
- import torch
17
  from yolov5.models.yolo import Model
 
18
 
19
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
20
  model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
21
  model.eval()
22
 
23
- from yolov5.utils.general import non_max_suppression
24
-
25
  def preprocess_image(image):
26
  image_tensor = F.to_tensor(image)
27
  return image_tensor.unsqueeze(0).to(device)
 
 
 
 
 
1
  import torch
2
  import torchvision
3
  from torchvision.transforms import functional as F
 
5
  import cv2
6
  import gradio as gr
7
 
 
 
 
 
 
8
  from yolov5.models.yolo import Model
9
+ from yolov5.utils.general import non_max_suppression
10
 
11
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12
  model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
13
  model.eval()
14
 
 
 
15
  def preprocess_image(image):
16
  image_tensor = F.to_tensor(image)
17
  return image_tensor.unsqueeze(0).to(device)