Update app.py

app.py

@@ -109,14 +109,14 @@ def get_grad(img):
     if y_pred[0] > 0.5: pred_class = "cat"
     else: pred_class = "dog"
         
-    text = "Raw Score: " + str(y_pred) + "\nClassification: " + pred_class
+    text = "Raw Score: " + str(y_pred[0]) + "\nClassification: " + pred_class
     return grad_img, text
 
 demo = gr.Interface(
   fn = get_grad,
   inputs = gr.Image(type = "pil", shape = (224,224)),
   # outputs = [gr.outputs.Label(num_top_classes = 2, label = 'Classifiaction'), gr.Textbox('infer_time', label = 'Inference Time(ms)')]
-  outputs = [gr.Image(type = "numpy"), gr.Textbox(label = 'Prediction (threshold: 0.5)')]
+  outputs = [gr.Image(type = "numpy", width = 160, height = 160), gr.Textbox(label = 'Prediction', info = '(threshold: 0.5)')]
    )
 demo.launch()