Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import torch.nn.functional as nnf
 import gradio as gr
 import numpy as np
+import pandas as pd
 import json
 
 
@@ -25,9 +26,9 @@ classes.sort()
 labels = classes
 
 def classify_image(inp):
-  print(inp.shape)
+  #print(inp.shape)
   inp = inp.astype(np.uint8).reshape((-1, 3, 300, 300))
-  print(inp.shape)
+  #print(inp.shape)
   inp = torch.from_numpy(inp).float()
   #confidences = model(inp)
 
@@ -39,6 +40,7 @@ def classify_image(inp):
   preds = nnf.softmax(preds, dim=0)
   preds = [pred.cpu() for pred in preds]
   preds = [float(pred.detach()) for pred in preds]
+  print(pd.Series(preds).describe())
 
   #confidences_dict = {classes[i]: float(confidences.data[0][i]) for i in range(len(confidences.data[0]))}
   confidences_dict = {classes[i]: float(preds[i]) for i in range(len(preds))}