Felguk commited on
Commit
29dd7fa
·
verified ·
1 Parent(s): fab6d88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -12,6 +12,7 @@ import torchvision
12
  from torchvision.models.detection import fasterrcnn_resnet50_fpn
13
  from torchvision.transforms import functional as F
14
  import io
 
15
 
16
  # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
17
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -81,10 +82,11 @@ def convert_image(image, target_format):
81
  if image is None:
82
  return None
83
  img = Image.open(image.name)
84
- buffer = io.BytesIO()
85
- img.save(buffer, format=target_format)
86
- buffer.seek(0)
87
- return buffer
 
88
 
89
  # Функция для определения языка текста
90
  def detect_language(text):
 
12
  from torchvision.models.detection import fasterrcnn_resnet50_fpn
13
  from torchvision.transforms import functional as F
14
  import io
15
+ import tempfile
16
 
17
  # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
18
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
82
  if image is None:
83
  return None
84
  img = Image.open(image.name)
85
+
86
+ # Создаем временный файл
87
+ with tempfile.NamedTemporaryFile(delete=False, suffix=f".{target_format.lower()}") as tmp_file:
88
+ img.save(tmp_file, format=target_format)
89
+ return tmp_file.name
90
 
91
  # Функция для определения языка текста
92
  def detect_language(text):