Update app.py

app.py

@@ -165,20 +165,24 @@ with st.sidebar.expander("tasks documentation"):
 
 # functions ########################################
 cwd = os.getcwd()
-def get_input(upload,text):
+def get_input(upload,url,text):
     if upload is not None:
         return upload
-    elif:
-        return text
-    return None
+    else:
+        if url:
+            print(url)
+            return url
+        elif text:
+            return text
+    return None  # Default return if neither upload nor url is provided
 
 def display_inputs(task):
-    if dict_hg_tasks_params[task]["input"] == "upload":
-        return st.file_uploader("Choose a file"),""
+    if dict_hg_tasks_params[task]["input"] == "upload,url":
+        return st.file_uploader("Choose a file"),st.text_input("or enter a file url"),""
     elif dict_hg_tasks_params[task]["input"] == "text":
-        return None,st.text_input("Enter a text")
+        return None,"",st.text_input("Enter a text")
     else:
-        return None,""
+        return None,"",""
     
 def display_prompt(task):
     if dict_hg_tasks_params[task]["prompt"] is True:
@@ -196,14 +200,14 @@ if selected_task :
     response = None
     task = dict_hg_tasks[selected_task]
     if model:
-        client = InferenceClient(model=model,token=hg_token)
+        client = InferenceClient(model=model)
     else:
-        client = InferenceClient(token=hg_token)    
-    uploaded_input,text_input = display_inputs(task)
+        client = InferenceClient()    
+    uploaded_input,url_input,text_input = display_inputs(task)
     prompt_input = display_prompt(task)
     context_input = display_context(task)
-    if get_input(uploaded_input,text_input):
-        input = get_input(uploaded_input,text_input) 
+    if get_input(uploaded_input,url_input,text_input):
+        input = get_input(uploaded_input,url_input,text_input) 
         response = getattr(client, task)(input)
     elif prompt_input:
         if context_input is not None:
@@ -228,5 +232,4 @@ if selected_task :
             elif dict_hg_tasks_params[task]["output"] == "image":
                 response.save(os.path.join(cwd,"generated_image.png"))
                 image = Image.open(os.path.join(cwd,"generated_image.png"))
-                st.image(image)
-
+                st.image(image)