Update app.py

app.py

@@ -9,9 +9,9 @@ api_key = os.environ['API_KEY']
 API_URL = "https://api-inference.huggingface.co/models/Hrishikesh332/autotrain-meme-classification-42897109437"
 headers = {"Authorization": f"Bearer {api_key}"}
 
-def query(filename):
-    with open(filename, "rb") as f:
-        data = f.read()
+def query(data : bytes):
+    # with open(filename, "rb") as f:
+    #     data = f.read()
     response = requests.post(API_URL, headers=headers, data=data)
     return response.json()
 
@@ -36,8 +36,8 @@ img = st.file_uploader("Choose an image", type=["jpg", "jpeg", "png"])
     
 if img is not None:
         # image = Image.open(BytesIO(img.read()))
-        image = process_file(img)
-        output = query(image)
+        data = img.read()
+        output = query(data)
         st.write("Predicted Output:", output)