seawolf2357 commited on
Commit
efe93ef
Β·
verified Β·
1 Parent(s): aa16f61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -23,11 +23,7 @@ def generate_image(prompt, negative_prompt):
23
  "Authorization": f"Bearer {os.getenv('HF_TOKEN')}"
24
  }
25
  data = {
26
- "inputs": {
27
- "prompt": prompt,
28
- "negative_prompt": negative_prompt,
29
- "num_inference_steps": 50
30
- }
31
  }
32
  api_url = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-Final"
33
  response = requests.post(api_url, headers=headers, json=data)
@@ -38,7 +34,6 @@ def generate_image(prompt, negative_prompt):
38
  logging.error(f"API μš”μ²­μ— μ‹€νŒ¨ν–ˆμŠ΅λ‹ˆλ‹€: {response.text}")
39
  return None
40
 
41
-
42
  # ν”„λ‘¬ν”„νŠΈ λ²ˆμ—­ ν•¨μˆ˜
43
  def translate_prompt(prompt):
44
  logging.debug(f'ν”„λ‘¬ν”„νŠΈ λ²ˆμ—­ 쀑: {prompt}')
 
23
  "Authorization": f"Bearer {os.getenv('HF_TOKEN')}"
24
  }
25
  data = {
26
+ "inputs": prompt + " " + negative_prompt # ν”„λ‘¬ν”„νŠΈμ™€ λ„€κ±°ν‹°λΈŒ ν”„λ‘¬ν”„νŠΈλ₯Ό ν•˜λ‚˜μ˜ λ¬Έμžμ—΄λ‘œ κ²°ν•©
 
 
 
 
27
  }
28
  api_url = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-Final"
29
  response = requests.post(api_url, headers=headers, json=data)
 
34
  logging.error(f"API μš”μ²­μ— μ‹€νŒ¨ν–ˆμŠ΅λ‹ˆλ‹€: {response.text}")
35
  return None
36
 
 
37
  # ν”„λ‘¬ν”„νŠΈ λ²ˆμ—­ ν•¨μˆ˜
38
  def translate_prompt(prompt):
39
  logging.debug(f'ν”„λ‘¬ν”„νŠΈ λ²ˆμ—­ 쀑: {prompt}')