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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -12,7 +12,6 @@ import torchvision
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,11 +81,10 @@ def convert_image(image, target_format):
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):
 
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
  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):