Update app.py

app.py

@@ -3,6 +3,7 @@ from huggingface_hub import InferenceClient, AsyncInferenceClient
 from PIL import Image
 from pathlib import Path
 import os, subprocess
+import requests
 
 st.set_page_config(page_title='HG Inference Client Demo',layout="wide")
 # Cache the header of the app to prevent re-rendering on each load
@@ -223,7 +224,10 @@ if selected_task :
                 st.audio(os.path.join(cwd,"audio.flac"))
         with col2:
             if dict_hg_tasks_params[task]["output"] == "image,text":
-                image = Image.open(input)
+                if not(isinstance(input, str)):
+                    image = Image.open(input)
+                else:
+                    image = Image.open(requests.get(input, stream=True).raw)
                 st.image(image)
             elif dict_hg_tasks_params[task]["output"] == "image":
                 response.save(os.path.join(cwd,"generated_image.png"))