forgot comma

app.py

@@ -19,6 +19,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
+
 def load_checkpoint(filepath):
     """Builds PyTorch Model from saved model
     Returns built model
@@ -88,9 +89,11 @@ class Network(nn.Module):
         x = self.output(x)
 
         return F.log_softmax(x, dim=1)
-    
+
+
 model = load_checkpoint("flower_inference_model.pth")
-    
+
+
 def process_image(img_path):
     """Scales, crops, and normalizes a PIL image for a PyTorch model,
     returns a Numpy array
@@ -154,6 +157,11 @@ def process_image(img_path):
 
     return np_image
 
+
+with open("cat_to_name.json", "r") as f:
+    cat_to_name = json.load(f)
+
+
 def predict(image_path, model=model, category_names=cat_to_name, topk=5):
     """Predict the class (or classes) of an image using a trained deep learning model.
     Arguments
@@ -216,10 +224,11 @@ def predict(image_path, model=model, category_names=cat_to_name, topk=5):
         plt.show()
 
         return fig
-    
+
+
 gr.Interface(
     predict,
     inputs=gr.inputs.Image(label="Upload a flower image", type="filepath"),
-    outputs=gr.outputs.Label(num_top_classes=5),
-    title="What kind of flower is this?"
-).launch()
+    outputs=gr.Plot(label="Plot"),
+    title="What kind of flower is this?",
+).launch()