vilarin commited on
Commit
60bcc4a
1 Parent(s): 98ca206

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,7 @@ footer {
30
  # Initialize the tokenizer
31
  tokenizer = AutoTokenizer.from_pretrained(model)
32
 
33
- llm = LLM(model=model)
34
 
35
  @spaces.GPU
36
  def generate(message, history, system, max_tokens, temperature, top_p, top_k, penalty):
@@ -58,6 +58,7 @@ def generate(message, history, system, max_tokens, temperature, top_p, top_k, pe
58
  eos_token_id=[151645,151643],
59
  )
60
  # generate outputs
 
61
  outputs = llm.generate([text], sampling_params)
62
 
63
  # Print the outputs.
 
30
  # Initialize the tokenizer
31
  tokenizer = AutoTokenizer.from_pretrained(model)
32
 
33
+
34
 
35
  @spaces.GPU
36
  def generate(message, history, system, max_tokens, temperature, top_p, top_k, penalty):
 
58
  eos_token_id=[151645,151643],
59
  )
60
  # generate outputs
61
+ llm = LLM(model=model)
62
  outputs = llm.generate([text], sampling_params)
63
 
64
  # Print the outputs.