update

app.py

@@ -30,7 +30,6 @@ def transform_image(img_sample):
 
 def predict(Image):
     tranformed_img = transform_image(Image)
-    """
     model.eval()
     tranformed_img = transform_image(Image)
     img = torch.from_numpy(tranformed_img)
@@ -42,18 +41,17 @@ def predict(Image):
     for cat, value in zip(category, grade):
         output_dict[cat] = value.item()
     return output_dict
-    """
-    return tranformed_img
+
 
 
 
 image = gr.Image(type="pil")#shape=(224, 224), image_mode="RGB")
-#label = gr.Label(label="Grade")
+label = gr.Label(label="Grade")
 
 demo = gr.Interface(
     fn=predict,
     inputs=image,
-    outputs=image,#label,
+    outputs=label,
     #examples=["examples/0.png", "examples/1.png", "examples/2.png", "examples/3.png", "examples/4.png"]
     )