Rooni commited on
Commit
fa58932
·
verified ·
1 Parent(s): 585a967

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -11
app.py CHANGED
@@ -25,29 +25,42 @@ def ask_openai_with_image(instruction, image):
25
  emj = random.choice(emojis)
26
  raise gr.Error(f"{emj} Заполни пожалуйста хотя бы одно поле")
27
 
28
- payload = {
 
 
 
 
29
  "model": "gpt-4-vision-preview",
30
  "messages": [
31
  {
32
  "role": "system",
33
  "content": start,
34
  },
 
 
 
35
  {
36
  "role": "user",
37
  "content": instruction if instruction else "",
38
  }
39
  ],
40
  "max_tokens": 4095,
41
- }
42
-
43
- if image is not None:
44
- # Кодируем загруженное изображение в base64
45
- base64_image = encode_image_to_base64(image)
46
- # Добавляем изображение в запрос
47
- payload["messages"].append({
48
- "role": "user",
49
- "content": f"data:image/jpeg;base64,{base64_image}"
50
- })
 
 
 
 
 
 
51
 
52
  # Заголовки для запроса
53
  headers = {
 
25
  emj = random.choice(emojis)
26
  raise gr.Error(f"{emj} Заполни пожалуйста хотя бы одно поле")
27
 
28
+
29
+ if image is not None:
30
+ # Кодируем загруженное изображение в base64
31
+ base64_image = encode_image_to_base64(image)
32
+ payload = {
33
  "model": "gpt-4-vision-preview",
34
  "messages": [
35
  {
36
  "role": "system",
37
  "content": start,
38
  },
39
+ {
40
+ "role": "user",
41
+ "content": f"data:image/jpeg;base64,{base64_image}"},
42
  {
43
  "role": "user",
44
  "content": instruction if instruction else "",
45
  }
46
  ],
47
  "max_tokens": 4095,
48
+ }
49
+ else:
50
+ payload = {
51
+ "model": "gpt-4-vision-preview",
52
+ "messages": [
53
+ {
54
+ "role": "system",
55
+ "content": start,
56
+ },
57
+ {
58
+ "role": "user",
59
+ "content": instruction if instruction else "",
60
+ }
61
+ ],
62
+ "max_tokens": 4095,
63
+ }
64
 
65
  # Заголовки для запроса
66
  headers = {