Spaces:

MesonWarrior
/

vk

Sleeping

MesonWarrior commited on Apr 8, 2023

Commit

4ffb6ee

•

1 Parent(s): 91f0ca0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,8 +13,8 @@ def generate_text(
 ):
     pipe = pipeline(
         'text-generation',
-        model='MesonWarrior/gpt2-bugro',
-        tokenizer='MesonWarrior/gpt2-bugro',
         min_length=min_length,
         max_length=max_length,
         use_auth_token=True

 ):
     pipe = pipeline(
         'text-generation',
+        model='MesonWarrior/gpt2-vk-bugro',
+        tokenizer='MesonWarrior/gpt2-vk-bugro',
         min_length=min_length,
         max_length=max_length,
         use_auth_token=True