Update app.py

app.py

@@ -3,8 +3,6 @@ import os
 import requests
 import io
 from PIL import Image
-from IPython.display import Audio, display
-from freeGPT import Client
 
 api_token = os.environ.get("API_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/facebook/musicgen-small"
@@ -16,12 +14,15 @@ st.write("Music generator using Facebook MusicGen, ChatGPT3 and Blip image capti
 img_prompt = st.file_uploader("Upload Image", type=["jpeg", "jpg", "png"])
 subm_btn = st.button("✨ Generate")
 
-if subm_btn:
-    def query(filename):
-        with open(filename, "rb") as f:
-            data = f.read()
-        response = requests.post(API_URL, headers=headers, data=data)
+if subm_btn and img_prompt is not None:
+    def query(image_bytes):
+        response = requests.post(API_URL_IMG, headers=headers, data=image_bytes)
         return response.json()
 
-    output = query(img_prompt)
-    st.info(f"Generated Prompt for input image: {output}")
+    image = Image.open(img_prompt)
+    image_bytes = io.BytesIO()
+    image.save(image_bytes, format=image.format)
+    image_bytes = image_bytes.getvalue()
+
+    output = query(image_bytes)
+    st.info(f"Generated Prompt for input image: {output}")