Trying to resolve

app.py

@@ -10,7 +10,7 @@ def nowcasting(file, cfg_str, ensemble_no):
     # Model Setup
     Forecastor = InferenceHub(
       model_config=STLDM_HKO, 
-      cfg_str=cfg_str,
+      cfg_str=cfg_str if cfg_str > 0 else None,
       model_type='HF'
     )
 

utilspp.py

@@ -144,23 +144,39 @@ def gradio_gif(sequences, T):
     fig, axes = plt.subplots(nrows=1, ncols=horizontal, figsize=(fig_size*horizontal, fig_size), tight_layout=True)
     plt.subplots_adjust(hspace=0.0, wspace=0.0) # tight layout
     plt.setp(axes, xticks=[], yticks=[])
-    for i, sequence in enumerate(sequences.values()):
-        axes[i].set_xticks([])
-        axes[i].set_yticks([])
-        axes[i].set_xlabel(f'Ensemble {i+1}', fontsize=12)
-        frame = sequence[0].squeeze()
-        im = axes[i].imshow(frame*255, cmap=ListedColormap(VIL_COLORS), \
-                                norm=BoundaryNorm(VIL_LEVELS, ListedColormap(VIL_COLORS).N), animated=True)
+
+    if horizontal == 1:
+        for i, sequence in enumerate(sequences.values()):
+            axes.set_xticks([])
+            axes.set_yticks([])
+            axes.set_xlabel(f'Ensemble {i+1}', fontsize=12)
+            frame = sequence[0].squeeze()
+            im = axes.imshow(frame*255, cmap=ListedColormap(VIL_COLORS), \
+                                    norm=BoundaryNorm(VIL_LEVELS, ListedColormap(VIL_COLORS).N), animated=True)
+    else:
+        for i, sequence in enumerate(sequences.values()):
+            axes[i].set_xticks([])
+            axes[i].set_yticks([])
+            axes[i].set_xlabel(f'Ensemble {i+1}', fontsize=12)
+            frame = sequence[0].squeeze()
+            im = axes[i].imshow(frame*255, cmap=ListedColormap(VIL_COLORS), \
+                                    norm=BoundaryNorm(VIL_LEVELS, ListedColormap(VIL_COLORS).N), animated=True)
 
     title = fig.suptitle('', y=0.9, x=0.505, fontsize=16) # Initialize an empty super title
 
     fig.colorbar(im)
 
     def animate(t):
-        for i, sequence in enumerate(sequences.values()):
-            frame = sequence[t].squeeze()
-            im = axes[i].imshow(frame*255, cmap=ListedColormap(VIL_COLORS), \
-                                          norm=BoundaryNorm(VIL_LEVELS, ListedColormap(VIL_COLORS).N), animated=True)
+        if horizontal == 1:
+           for i, sequence in enumerate(sequences.values()):
+                frame = sequence[t].squeeze()
+                im = axes.imshow(frame*255, cmap=ListedColormap(VIL_COLORS), \
+                                                norm=BoundaryNorm(VIL_LEVELS, ListedColormap(VIL_COLORS).N), animated=True)
+        else:
+            for i, sequence in enumerate(sequences.values()):
+                frame = sequence[t].squeeze()
+                im = axes[i].imshow(frame*255, cmap=ListedColormap(VIL_COLORS), \
+                                            norm=BoundaryNorm(VIL_LEVELS, ListedColormap(VIL_COLORS).N), animated=True)
         plt.subplots_adjust(hspace=0.0, wspace=0.0) # tight layout
         
         title.set_text(f'$t + {t}$') # update the title text