added no class detected case

app.py

@@ -113,7 +113,11 @@ def do_inference(img):
       probs = probs.cpu().numpy()[0]
       probs = probs[output]
       labels = np.array(class_names)[output]
-      return {id_to_name[labels[i]]: round(float(probs[i]),2) for i in range(len(labels))}
+      
+      if(probs[0] < 0.5):
+          return {"No butterfly":1.0}
+      else:  
+        return {id_to_name[labels[i]]: round(float(probs[i]),2) for i in range(len(labels))}
 
 im = gr.inputs.Image(shape=(512, 512), image_mode='RGB', 
                          invert_colors=False, source="upload",