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 =
|
32 |
return result
|
33 |
|
34 |
|
|
|
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 |
|