fix none type error

GooeyAI · Jul 5, 2024 · 4a0f1f5 · 4a0f1f5
1 parent b67c8a4
commit 4a0f1f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recipes/CompareUpscaler.py b/recipes/CompareUpscaler.py
@@ -183,10 +183,10 @@ def related_workflows(self) -> list:
 
 def _render_outputs(state):
     for key in state.get("selected_models") or []:
-        img = state.get("output_images", {}).get(key)
+        img = (state.get("output_images") or {}).get(key)
         if img:
             st.image(img, caption=UpscalerModels[key].label, show_download_button=True)
 
-        vid = state.get("output_videos", {}).get(key)
+        vid = (state.get("output_videos") or {}).get(key)
         if vid:
             st.video(vid, caption=UpscalerModels[key].label, show_download_button=True)