ported to chatgpt api

app.py

@@ -78,15 +78,16 @@ def predict(img, prompt):
         # Get the file ID from the upload response
     file_id = upload_response['id']
     messages = [
-        {"role": "system", "content": prompt},
-        {"role": "user", "content": f"File ID: {file_id}"}
+        {"role": "system", "content": prompt + "respond with JSON only"},
+        {"role": "user", "content": f"Image (base64): {image_bytes}"}
     ]
 
     # Make API call
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-4-vision-preview",
             messages=messages,
+             files=[("file", file_id)]
         )
         return response['choices'][0]['message']['content']
     except Exception as e: