Update app.py

app.py

@@ -21,7 +21,7 @@ torch.hub.download_url_to_file(
 
 # Model
 # model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # force_reload=True to update
-model = torch.hub.load('./yolov5', 'custom', path='plate.pt', source="local") 
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='plate.pt', source="local") 
 
 
 def yolo(im):
@@ -36,7 +36,6 @@ def yolo(im):
     res = json.loads(df)
 
     return [Image.fromarray(results.ims[0]), res]
-    # return [Image.fromarray(results.ims[0])]
 
 
 
@@ -45,11 +44,10 @@ def yolo(im):
 inputs = gr.inputs.Image(type='pil', label="Original Image")
 outputs = [gr.outputs.Image(type="pil", label="Output Image"),
     gr.outputs.JSON(label="Output JSON")]
-# outputs = gr.outputs.Image(type="pil", label="Output Image")
 
 title = "TW_plate_number"
 description = "TW_plate_number"
-
+# article = "<p style='text-align: center'>TW_plate_number <a href=\"http://codh.rois.ac.jp/char-shape/\">日本古典籍くずし字データセット</a>.</p>"
 
 examples = [['1.jpg'], ['2.jpg'], ['3.jpg'], ['4.jpg'], ['5.jpg'], ['6.jpg']]
-gr.Interface(yolo, inputs, outputs, title=title, description=description, examples=examples, theme="huggingface").launch(enable_queue=True) 
+gr.Interface(yolo, inputs, outputs, title=title, description=description, examples=examples, theme="huggingface").launch(enable_queue=True) # cache_examples=True,