minoD commited on
Commit
4949853
·
1 Parent(s): 7a8d345

remove cuda

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,7 +35,7 @@ def generate2(F=None, maxTokens=256):
35
  input_ids = tokenizer(prompt,
36
  return_tensors="pt",
37
  truncation=True,
38
- add_special_tokens=False).input_ids.cuda()
39
  outputs = model.generate(
40
  input_ids=input_ids,
41
  max_new_tokens=maxTokens,
@@ -77,4 +77,4 @@ iface = gr.Interface(
77
  )
78
 
79
  if __name__ == "__main__":
80
- iface.launch()
 
35
  input_ids = tokenizer(prompt,
36
  return_tensors="pt",
37
  truncation=True,
38
+ add_special_tokens=False).input_ids
39
  outputs = model.generate(
40
  input_ids=input_ids,
41
  max_new_tokens=maxTokens,
 
77
  )
78
 
79
  if __name__ == "__main__":
80
+ iface.launch(share=True)