plot modifications

app.py

@@ -175,7 +175,7 @@ def plot_input(input_image: list, origin='lower'):
     plt.rcParams['xtick.labelsize'] = ticks
     plt.rcParams['ytick.labelsize'] = ticks
     
-    input_fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(10, 8))
+    input_fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(6, 5))
     
     im0 = ax.imshow(input_image, cmap=cmap,
                      origin=origin)    
@@ -295,7 +295,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     model_url = cached_download(hf_hub_url(model_path + model_loading_name, filename="pytorch_model.bin"))
     # print(model_url)
     
-    loaded = torch.load(model_url, map_location='cpu', )
+    loaded = torch.load(model_url, map_location='cpu',)
     # print(loaded.keys())
     
     model.load_state_dict(loaded['state_dict'])
@@ -335,7 +335,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     
     print("Plotting")
     
-    origin = 'upper'
+    origin = 'lower'
     
     # plot input image
     input_fig = plot_input(input_image, origin=origin)