Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-# import tensorflow as tf
+import tensorflow as tf
 import os
 import keras
 import gdown
@@ -22,21 +22,18 @@ labels = {
     9: "truck",
 }
 
-full_path = ""
 # Download the model file
 def download_model():
-    url = "https://drive.google.com/uc?id=12700bE-pomYKoVQ214VrpBoJ7akXcTpL"
+    #url = "https://drive.google.com/uc?id=12700bE-pomYKoVQ214VrpBoJ7akXcTpL"
+    url = "https://drive.google.com/uc?id=1fclkgIgUo26g014beN8UhCJ2TfvGG2sG"
     output = "modelV2Lmixed.keras"
     gdown.download(url, output, quiet=False)
-    full_path = os.path.abspath(output)
-    #return output
-    return full_path
+    return output
 
-#model_file = download_model()
-download_model()
+model_file = download_model()
 
 # Load the model
-model = keras.saving.load_model(full_path)
+model = tf.keras.models.load_model(model_file)
 
 # Perform image classification for single class output
 # def predict_class(image):