sam / request.py
sezer91's picture
ds
725e7c2
raw
history blame
1.28 kB
import requests
from PIL import Image
import base64
from io import BytesIO
url = "https://sezer91-sam.hf.space/segment/"
file_path = "img.jpeg"
try:
# Görüntü dosyasını kontrol et
img = Image.open(file_path)
if img.size[0] < 64 or img.size[1] < 64:
raise ValueError(f"Görüntü boyutu çok küçük: {img.size[0]}x{img.size[1]}. Minimum 64x64 piksel olmalı.")
print(f"Görüntü boyutu: {img.size[0]}x{img.size[1]}")
with open(file_path, "rb") as file:
files = {"file": file}
response = requests.post(url, files=files)
if response.status_code == 200:
result = response.json()
print("Başarılı! Maske alındı.")
# Base64'ü PNG olarak kaydet
base64_string = result["mask"].split(",")[1] # "data:image/png;base64," kısmını atla
img_data = base64.b64decode(base64_string)
img = Image.open(BytesIO(img_data))
img.save("output_mask.png")
print("Maske 'output_mask.png' olarak kaydedildi.")
else:
print(f"Hata: {response.status_code}, {response.text}")
except FileNotFoundError:
print(f"Hata: {file_path} dosyası bulunamadı.")
except ValueError as ve:
print(f"Hata: {str(ve)}")
except Exception as e:
print(f"Hata: {str(e)}")