Spaces:
Sleeping
Sleeping
os and env
Browse files- app.py +2 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
from transformers import pipeline, set_seed
|
3 |
|
4 |
|
@@ -7,7 +8,7 @@ set_seed(42)
|
|
7 |
|
8 |
|
9 |
def Bemenet(bemenet):
|
10 |
-
return generator(SECRET_PROMPT + " " + bemenet, max_length=100, num_return_sequences=2)[1]['generated_text']
|
11 |
|
12 |
|
13 |
interface = gr.Interface(fn=Bemenet,
|
|
|
1 |
import gradio as gr
|
2 |
+
import os
|
3 |
from transformers import pipeline, set_seed
|
4 |
|
5 |
|
|
|
8 |
|
9 |
|
10 |
def Bemenet(bemenet):
|
11 |
+
return generator(os.environ['SECRET_PROMPT'] + " " + bemenet, max_length=100, num_return_sequences=2)[1]['generated_text']
|
12 |
|
13 |
|
14 |
interface = gr.Interface(fn=Bemenet,
|
requirements.txt
CHANGED
@@ -1,2 +1,3 @@
|
|
1 |
transformers
|
2 |
-
tensorflow
|
|
|
|
1 |
transformers
|
2 |
+
tensorflow
|
3 |
+
os
|