truongghieu commited on
Commit
17018a6
1 Parent(s): e544d42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -2,6 +2,11 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM , GenerationConfig
3
 
4
 
 
 
 
 
 
5
  generation_config = GenerationConfig(
6
  penalty_alpha=0.6,
7
  do_sample = True,
@@ -12,11 +17,6 @@ generation_config = GenerationConfig(
12
  pad_token_id=tokenizer.eos_token_id
13
  )
14
 
15
- tokenizer = AutoTokenizer.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
16
- model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
17
-
18
-
19
-
20
 
21
  # Define a function that takes a text input and generates a text output
22
  def generate_text(text):
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM , GenerationConfig
3
 
4
 
5
+
6
+
7
+ tokenizer = AutoTokenizer.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
8
+ model = AutoModelForCausalLM.from_pretrained("truongghieu/deci-finetuned", trust_remote_code=True)
9
+
10
  generation_config = GenerationConfig(
11
  penalty_alpha=0.6,
12
  do_sample = True,
 
17
  pad_token_id=tokenizer.eos_token_id
18
  )
19
 
 
 
 
 
 
20
 
21
  # Define a function that takes a text input and generates a text output
22
  def generate_text(text):