Hugging Face's logo

Spaces:
panda1835
/
vnturtle-detect-api
Runtime error

App Files Community
panda1835 commited on
Commit
c5343cf
·
1 Parent(s): 60e30ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,5 +1,4 @@
1
- from keras.models import load_model
2
- from PIL import Image, ImageOps
3
  import numpy as np
4
  import gradio as gr
5
  import pandas as pd
@@ -7,7 +6,6 @@ import torch
7
  import json
8
 
9
  # Load the model
10
- classify_model = load_model('keras_model.h5')
11
  detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)
12
 
13
  def detect(image):
 
1
+ from PIL import Image
 
2
  import numpy as np
3
  import gradio as gr
4
  import pandas as pd
 
6
  import json
7
 
8
  # Load the model
 
9
  detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)
10
 
11
  def detect(image):