Spaces:

MesonWarrior
/

vk

Sleeping

MesonWarrior commited on Apr 8, 2023

Commit

d0bec4c

•

1 Parent(s): 1abdf30

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,8 @@
 import gradio as gr
 from transformers import pipeline
 def generate_text(
     model,
@@ -14,7 +17,7 @@ def generate_text(
         tokenizer='MesonWarrior/gpt2-bugro',
         min_length=min_length,
         max_length=max_length,
-        use_auth_token="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"
     )
     return pipe(text)[0]['generated_text']

 import gradio as gr
 from transformers import pipeline
+from huggingface-hub import login
+login("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")
 def generate_text(
     model,
         tokenizer='MesonWarrior/gpt2-bugro',
         min_length=min_length,
         max_length=max_length,
+        use_auth_token=True
     )
     return pipe(text)[0]['generated_text']