Anish13 commited on
Commit
f030c04
·
1 Parent(s): 24a5b1e
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -181,7 +181,7 @@ context = torch.zeros((1, 1), dtype=torch.long, device=device)
181
  def greet(start_character, number_of_tokens):
182
  context[0][0] = encode(start_character)[0]
183
  max_new_tokens = number_of_tokens
184
- return decode(model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
185
 
186
  iface = gr.Interface(fn=greet, inputs=["text", "number"], outputs="text")
187
- iface.launch()
 
181
  def greet(start_character, number_of_tokens):
182
  context[0][0] = encode(start_character)[0]
183
  max_new_tokens = number_of_tokens
184
+ return decode(model.generate(context, max_new_tokens=int(max_new_tokens))[0].tolist())
185
 
186
  iface = gr.Interface(fn=greet, inputs=["text", "number"], outputs="text")
187
+ iface.launch(share=True)