v2

app.py

@@ -8,7 +8,7 @@ import numpy as np
 model = from_pretrained_keras("yusyel/fishv2")
 
 
-class_names=["Black Sea Sprat",
+CLASS=["Black Sea Sprat",
  "Gilt-Head Bream",
  "Hourse Mackerel",
  "Red Mullet",
@@ -34,7 +34,7 @@ def predict(img):
     pred = model.predict(img)
     pred = np.squeeze(pred).astype(float)
     print(pred)
-    return dict(zip(class_names, pred))
+    return dict(zip(CLASS, pred))
 
 
 demo = gr.Interface(