fittar commited on
Commit
dfab003
1 Parent(s): 8dcc59b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,7 +28,7 @@ tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
28
  tokenizer.pad_token = tokenizer.eos_token
29
 
30
  def generate(text):
31
- result =vipe_generate(text,model,tokenizer,do_sample=True,device=device)
32
  return result
33
 
34
 
@@ -53,4 +53,4 @@ demo = gr.Interface(
53
  theme="default"
54
  )
55
 
56
- demo.launch(share=True,enable_queue=True, debug=True)
 
28
  tokenizer.pad_token = tokenizer.eos_token
29
 
30
  def generate(text):
31
+ result =vipe_generate([text],model,tokenizer,do_sample=True,device=device)
32
  return result
33
 
34
 
 
53
  theme="default"
54
  )
55
 
56
+ demo.launch(enable_queue=True, debug=True)