small fix agen

app.py

@@ -33,7 +33,7 @@ class GradioApp:
         
         img = model_transforms[model_name](Image.open(img_file)).unsqueeze(0)
         with torch.inference_mode():
-            preds = torch.softmax(self.models[model_name](img)[0], dim=0)[0].cpu().numpy()
+            preds = torch.softmax(self.models[model_name](img)[0], dim=0).cpu().numpy()
         return dict(zip(self.classes, preds))
     
     def launch(self):