Rooni commited on
Commit
774fc84
·
1 Parent(s): 67da721

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,11 +4,12 @@ import os
4
  import base64
5
  from PIL import Image
6
  import numpy as np
 
7
 
8
  # Функция для обработки изображения и текста и отправки запроса к OpenAI
9
  def generate_text(image, prompt):
10
  # Конвертируем NumPy массив в объект Image
11
- image_pil = Image.fromarray(np.uint8(image)).convert('RGB')
12
 
13
  # Конвертируем изображение для отправки через API
14
  image_bytes = io.BytesIO()
 
4
  import base64
5
  from PIL import Image
6
  import numpy as np
7
+ import io # Добавляем необходимый импорт
8
 
9
  # Функция для обработки изображения и текста и отправки запроса к OpenAI
10
  def generate_text(image, prompt):
11
  # Конвертируем NumPy массив в объект Image
12
+ image_pil = Image.fromarray(image.astype('uint8'), 'RGB')
13
 
14
  # Конвертируем изображение для отправки через API
15
  image_bytes = io.BytesIO()