Spaces:

Mehulgarg
/

potato2.0

No application file

Mehulgarg commited on Dec 30, 2023

Commit

1993b58

•

1 Parent(s): 63767de

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -19,8 +19,7 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
-url="https://colab.research.google.com/drive/1dapIRAtaJdnM7xhMyjOpIp-8JEkOQ0iY#scrollTo=FyjH2NdRhkmc"
-MODEL = tf.keras.models.load_model("url")
 CLASS_NAMES = ["Early Blight", "Late Blight", "Healthy"]

     allow_methods=["*"],
     allow_headers=["*"],
 )
+MODEL = tf.keras.models.load_model("../saved_models/1")
 CLASS_NAMES = ["Early Blight", "Late Blight", "Healthy"]