csuvikv commited on
Commit
cfc1933
·
1 Parent(s): 93f8f0d

os and env

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. 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