shuhayas commited on
Commit
b34c8ea
·
verified ·
1 Parent(s): c946f95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -35,15 +35,25 @@ if uploaded_image is not None:
35
  vision_data = {
36
  'model': 'gpt-4-vision-preview',
37
  'messages': [
 
 
 
 
38
  {
39
  'role': 'user',
40
- 'content': [
41
- {'type': 'text', 'text': "What’s in this image?"},
42
- {'type': 'image_url', 'image_url': {'url': f'data:image/jpeg;base64,{img_b64}'}}
43
- ]
 
 
 
 
 
 
44
  }
45
  ],
46
- 'max_tokens': 300
47
  }
48
  response = requests.post(GPT4_VISION_API, headers=headers, json=vision_data)
49
  if response.status_code == 200:
 
35
  vision_data = {
36
  'model': 'gpt-4-vision-preview',
37
  'messages': [
38
+ {
39
+ 'role': 'system',
40
+ 'content': system_prompt
41
+ },
42
  {
43
  'role': 'user',
44
+ 'content': {
45
+ 'type': 'image',
46
+ 'data': {
47
+ 'image': f'data:image/jpeg;base64,{img_b64}'
48
+ }
49
+ }
50
+ },
51
+ {
52
+ 'role': 'assistant',
53
+ 'content': assistant_prompt
54
  }
55
  ],
56
+ 'max_tokens': 4000
57
  }
58
  response = requests.post(GPT4_VISION_API, headers=headers, json=vision_data)
59
  if response.status_code == 200: