Mars-Signvrse Marsrover1997 commited on
Commit
c7d4d14
·
verified ·
1 Parent(s): 37178e5

Update app.py (#1)

Browse files

- Update app.py (40ac7b6651657d18798553bc73d3e5a93dc75bd8)


Co-authored-by: Anthony Githinji Marugu <Marsrover1997@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ Sign Language Translator for converting sentences to Gloss. Glosses are written
33
  """
34
 
35
  def translate_to_gloss(input_text):
36
- inputs = tokenizer([tinyllama_prompt.format(input_text)], return_tensors="pt").to("cuda")
37
  with torch.no_grad():
38
  outputs = peft_model.generate(**inputs, max_new_tokens=64, use_cache=True)
39
  decoded_output = tokenizer.batch_decode(outputs)[0]
 
33
  """
34
 
35
  def translate_to_gloss(input_text):
36
+ inputs = tokenizer([tinyllama_prompt.format(input_text)], return_tensors="pt")
37
  with torch.no_grad():
38
  outputs = peft_model.generate(**inputs, max_new_tokens=64, use_cache=True)
39
  decoded_output = tokenizer.batch_decode(outputs)[0]