Update app.py
Browse files
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(
|
|
|
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)
|