Upload app.py

app.py

@@ -27,10 +27,7 @@ url = "https://drive.google.com/uc?id=12700bE-pomYKoVQ214VrpBoJ7akXcTpL"
 output = "modelV2Lmixed.keras"
 gdown.download(url, output, quiet=False)
 
-
-def load_model():
-    model = tf.keras.models.load_model("./modelV2Lmixed.keras")
-    return model
+model = tf.keras.models.load_model("./modelV2Lmixed.keras")
 
 
 def classify_image(image, model):
@@ -38,11 +35,8 @@ def classify_image(image, model):
     image = tf.image.resize(image, [32, 32])
     image = np.expand_dims(image, axis=0)
     prediction = model.predict(image)
-    confidences = {labels[i]: float(prediction[i]) for i in range(10)}
-    return confidences
-
-
-model = load_model()
+    #confidences = {labels[i]: float(prediction[i]) for i in range(10)}
+    return prediction
 
 
 gr.Interface(fn=classify_image,