kooldark commited on
Commit
fbe9a30
·
verified ·
1 Parent(s): 9d71f96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -6,10 +6,8 @@ import io
6
  from PIL import Image
7
  from mtranslate import translate # Import thư viện mtranslate
8
 
9
- # Tải biến môi trường từ file .env nếu có (chỉ dùng khi phát triển cục bộ)
10
  load_dotenv()
11
 
12
- # Lấy API Key từ biến môi trường
13
  api_key = os.getenv("HF_API_KEY")
14
  image_api_url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large-turbo"
15
  headers = {"Authorization": f"Bearer {api_key}"}
@@ -19,7 +17,6 @@ def query(payload):
19
  response = requests.post(image_api_url, headers=headers, json=payload)
20
  return response.content
21
 
22
- # Hàm dịch tiếng Việt sang tiếng Anh bằng mtranslate
23
  def translate_to_english(text):
24
  return translate(text, "en", "vi") # Dịch từ tiếng Việt sang tiếng Anh
25
 
@@ -33,7 +30,6 @@ def generate_image(prompt):
33
  image = Image.open(io.BytesIO(image_bytes))
34
  return image
35
 
36
- # Tạo giao diện Gradio
37
  iface = gr.Interface(
38
  fn=generate_image,
39
  inputs="text",
 
6
  from PIL import Image
7
  from mtranslate import translate # Import thư viện mtranslate
8
 
 
9
  load_dotenv()
10
 
 
11
  api_key = os.getenv("HF_API_KEY")
12
  image_api_url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large-turbo"
13
  headers = {"Authorization": f"Bearer {api_key}"}
 
17
  response = requests.post(image_api_url, headers=headers, json=payload)
18
  return response.content
19
 
 
20
  def translate_to_english(text):
21
  return translate(text, "en", "vi") # Dịch từ tiếng Việt sang tiếng Anh
22
 
 
30
  image = Image.open(io.BytesIO(image_bytes))
31
  return image
32
 
 
33
  iface = gr.Interface(
34
  fn=generate_image,
35
  inputs="text",