fantaxy commited on
Commit
6bdbd0a
โ€ข
1 Parent(s): 022b2fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -22
app.py CHANGED
@@ -31,28 +31,21 @@ def respond(
31
  messages.append({"role": "assistant", "content": assistant_msg})
32
  messages.append({"role": "user", "content": message})
33
 
34
- # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ์š”์ฒญ์„ API์— ์ „์†กํ•˜๊ณ  ๊ฒฐ๊ณผ๋ฅผ ์ฒ˜๋ฆฌํ•ฉ๋‹ˆ๋‹ค.
35
- try:
36
- result = client.predict(
37
- prompt=message,
38
- seed=123,
39
- randomize_seed=False,
40
- width=1024,
41
- height=576,
42
- guidance_scale=5,
43
- num_inference_steps=28,
44
- api_name="/infer_t2i"
45
- )
46
- if 'url' in result:
47
- return result['url']
48
- else:
49
- logging.error("์ด๋ฏธ์ง€ ์ƒ์„ฑ ์‹คํŒจ: %s", result.get('error', '์•Œ ์ˆ˜ ์—†๋Š” ์˜ค๋ฅ˜'))
50
- return "์ด๋ฏธ์ง€ ์ƒ์„ฑ์— ์‹คํŒจํ–ˆ์Šต๋‹ˆ๋‹ค."
51
- except Exception as e:
52
- logging.error("API ์š”์ฒญ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: %s", str(e))
53
- return f"์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}"
54
- finally:
55
- logging.info("์š”์ฒญ ์ฒ˜๋ฆฌ ์™„๋ฃŒ")
56
 
57
  theme = "Nymbo/Nymbo_Theme"
58
  css = """
 
31
  messages.append({"role": "assistant", "content": assistant_msg})
32
  messages.append({"role": "user", "content": message})
33
 
34
+ try:
35
+ result = client.predict(...)
36
+ if isinstance(result, dict) and 'url' in result:
37
+ return result['url']
38
+ else:
39
+ logging.error("๊ฒฐ๊ณผ ์ฒ˜๋ฆฌ ์‹คํŒจ: ๊ฒฐ๊ณผ๊ฐ€ ๋”•์…”๋„ˆ๋ฆฌ ํ˜•ํƒœ๊ฐ€ ์•„๋‹˜")
40
+ return "์ด๋ฏธ์ง€ ์ƒ์„ฑ์— ์‹คํŒจํ–ˆ์Šต๋‹ˆ๋‹ค."
41
+ except AttributeError as e:
42
+ logging.error("์†์„ฑ ์˜ค๋ฅ˜: %s", str(e))
43
+ return "๋‚ด๋ถ€ ์ฒ˜๋ฆฌ ์˜ค๋ฅ˜ ๋ฐœ์ƒ"
44
+ except Exception as e:
45
+ logging.error("API ์š”์ฒญ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: %s", str(e))
46
+ return f"์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}"
47
+ finally:
48
+ logging.info("์š”์ฒญ ์ฒ˜๋ฆฌ ์™„๋ฃŒ")
 
 
 
 
 
 
 
49
 
50
  theme = "Nymbo/Nymbo_Theme"
51
  css = """