Update app.py

app.py

@@ -26,17 +26,18 @@ def generate_image(prompt):
 
 if type == "Text":
     prompt = st.text_input("Enter prompt")
-    out = imquery(prompt)
-    st.image(out)
+    if prompt != None:
+        out = imquery(prompt)
+        st.image(out)
 else:
     prompt = st.file_uploader("Paste your audiofile", type=["mp3", "m4a", "wav"])
-
-    def query(filename):
-        with open(filename, "rb") as f:
-            data = f.read()
-        response = requests.post(API_URL_WH, headers=headers, data=data)
-        return response.json()
-        
-    output = query(prompt)
-    imoo = generate_image(output)
-    st.image(imoo)
+    if prompt != None:
+        def query(filename):
+            with open(filename, "rb") as f:
+                data = f.read()
+            response = requests.post(API_URL_WH, headers=headers, data=data)
+            return response.json()
+            
+        output = query(prompt)
+        imoo = generate_image(output)
+        st.image(imoo)