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)
|