alrab222 commited on
Commit
7f6a958
1 Parent(s): 561f241

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,11 +3,11 @@ import json
3
  import gradio as gr
4
  import yolov5
5
 
6
- model = yolov5.load('Cinderella.pt')
7
 
8
 
9
  def inference(gr_input):
10
- with open("json/classes.json", "r")as f:
11
  dic = json.load(f)
12
  predictions = model(gr_input).pred[0]
13
  name = [x for x in dic.values()]
 
3
  import gradio as gr
4
  import yolov5
5
 
6
+ model = yolov5.load('model/Cinderella.pt')
7
 
8
 
9
  def inference(gr_input):
10
+ with open("json/idol_classes.json", "r")as f:
11
  dic = json.load(f)
12
  predictions = model(gr_input).pred[0]
13
  name = [x for x in dic.values()]