vilarin commited on
Commit
32ff463
·
verified ·
1 Parent(s): 1ec2e60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ model = AutoModelForCausalLM.from_pretrained(
58
  )
59
  tokenizer = AutoTokenizer.from_pretrained(MODELS)
60
 
61
- @spaces.GPU
62
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int):
63
  print(f'message is - {message}')
64
  print(f'history is - {history}')
 
58
  )
59
  tokenizer = AutoTokenizer.from_pretrained(MODELS)
60
 
61
+ @spaces.GPU()
62
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int):
63
  print(f'message is - {message}')
64
  print(f'history is - {history}')