Spaces:
Sleeping
Sleeping
Commit
·
92149d8
1
Parent(s):
a8cfe5b
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,8 +1,5 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from transformers import pipeline
|
| 3 |
-
from huggingface_hub import login
|
| 4 |
-
|
| 5 |
-
login(token="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")
|
| 6 |
|
| 7 |
def generate_text(
|
| 8 |
model_name,
|
|
@@ -26,8 +23,7 @@ def generate_text(
|
|
| 26 |
model=model,
|
| 27 |
tokenizer=model,
|
| 28 |
min_length=min_length,
|
| 29 |
-
max_length=max_length
|
| 30 |
-
use_auth_token=True
|
| 31 |
)
|
| 32 |
|
| 33 |
return pipe(text, temperature=temperature, top_k=top_k, top_p=top_p, do_sample=True)[0]['generated_text']
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
| 3 |
|
| 4 |
def generate_text(
|
| 5 |
model_name,
|
|
|
|
| 23 |
model=model,
|
| 24 |
tokenizer=model,
|
| 25 |
min_length=min_length,
|
| 26 |
+
max_length=max_length
|
|
|
|
| 27 |
)
|
| 28 |
|
| 29 |
return pipe(text, temperature=temperature, top_k=top_k, top_p=top_p, do_sample=True)[0]['generated_text']
|