yuragoithf commited on
Commit
1d000d6
·
verified ·
1 Parent(s): 692a09b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -24,15 +24,16 @@ labels = {
24
  # Download the model file
25
  def download_model():
26
  url = "https://drive.google.com/uc?id=12700bE-pomYKoVQ214VrpBoJ7akXcTpL"
27
- output = "/home/user/modelV2Lmixed.keras"
28
  gdown.download(url, output, quiet=False)
29
- return output
 
30
 
31
  #model_file = download_model()
32
  download_model()
33
 
34
  # Load the model
35
- model = keras.saving.load_model("/home/user/modelV2Lmixed.keras")
36
 
37
  # Perform image classification for single class output
38
  # def predict_class(image):
 
24
  # Download the model file
25
  def download_model():
26
  url = "https://drive.google.com/uc?id=12700bE-pomYKoVQ214VrpBoJ7akXcTpL"
27
+ output = "modelV2Lmixed.keras"
28
  gdown.download(url, output, quiet=False)
29
+ #return output
30
+ return
31
 
32
  #model_file = download_model()
33
  download_model()
34
 
35
  # Load the model
36
+ model = keras.saving.load_model("modelV2Lmixed.keras")
37
 
38
  # Perform image classification for single class output
39
  # def predict_class(image):