Update app.py

app.py

@@ -2,6 +2,8 @@ import tf_keras as keras
 import gradio as gr
 import cv2
 import os
+from dotenv import load_dotenv
+load_dotenv()
 
 
 print(os.listdir('./model'))
@@ -12,7 +14,7 @@ def classify_image(img_dt):
   img_dt = cv2.resize(img_dt,(256,256))
   img_dt = img_dt.reshape((-1,256,256,3)) 
   prediction = used_model.predict(img_dt).flatten()
-  confidences = {new_classes[i]: float(prediction[i]) for i in range (4) }
+  confidences = {new_classes[i]: float(prediction[i]) for i in range (3) }
   return confidences