Update app.py

app.py

@@ -45,13 +45,13 @@ def predict(img) -> Tuple[Dict, float]:
     
     # Create a prediction label and prediction probability dictionary for each prediction class (this is the required format for Gradio's output parameter)        
     pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
-    pred_labels_and_prob_1={"Unknown Image" if v <0.75 else k:v for k,v in pred_labels_and_prob.items()}
+    #pred_labels_and_prob_1={"Unknown Image" if v <0.75 else k:v for k,v in pred_labels_and_prob.items()}
             
     # Calculate the prediction time
     pred_time = round(timer() - start_time, 5)
     
     # Return the prediction dictionary and prediction time 
-    return pred_labels_and_probs_1, pred_time
+    return pred_labels_and_probs, pred_time
 
 ### 4. Gradio app ###