Ping9gu commited on
Commit
4d20401
Β·
1 Parent(s): 909ec96

Simplify text generation parameters

Browse files
Files changed (2) hide show
  1. app.py +7 -11
  2. requirements.txt +1 -1
app.py CHANGED
@@ -36,17 +36,13 @@ def generate_diary(keywords):
36
  였늘의 일기:
37
  μ˜€λŠ˜μ€ """
38
 
39
- parameters = {
40
- "max_new_tokens": 768,
41
- "temperature": 0.88,
42
- "top_p": 0.95,
43
- "repetition_penalty": 1.35,
44
- "top_k": 50,
45
- "do_sample": True,
46
- "num_return_sequences": 1
47
- }
48
-
49
- response = client.text_generation(prompt, **parameters)
50
 
51
  if not response:
52
  return "일기 생성에 μ‹€νŒ¨ν–ˆμŠ΅λ‹ˆλ‹€. λ‹€μ‹œ μ‹œλ„ν•΄μ£Όμ„Έμš”."
 
36
  였늘의 일기:
37
  μ˜€λŠ˜μ€ """
38
 
39
+ # 더 λ‹¨μˆœν•œ νŒŒλΌλ―Έν„°λ‘œ μ‹œλ„
40
+ response = client.text_generation(
41
+ prompt,
42
+ max_new_tokens=768,
43
+ temperature=0.88,
44
+ do_sample=True
45
+ )
 
 
 
 
46
 
47
  if not response:
48
  return "일기 생성에 μ‹€νŒ¨ν–ˆμŠ΅λ‹ˆλ‹€. λ‹€μ‹œ μ‹œλ„ν•΄μ£Όμ„Έμš”."
requirements.txt CHANGED
@@ -4,7 +4,7 @@ requests==2.31.0
4
  python-dotenv==1.0.0
5
  werkzeug==2.3.7
6
  huggingface-hub>=0.17.0
7
- gradio==4.8.0
8
  torch
9
  transformers
10
  accelerate
 
4
  python-dotenv==1.0.0
5
  werkzeug==2.3.7
6
  huggingface-hub>=0.17.0
7
+ gradio==5.14.0
8
  torch
9
  transformers
10
  accelerate