csuvikv commited on
Commit
e9d6ce2
·
1 Parent(s): df2eb4a

interface textbox

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -11,11 +11,12 @@ generator = pipeline('text-generation', model='gpt2', return_full_text=False, to
11
  def Bemenet(bemenet):
12
  return generator(os.environ['SECRET_PROMPT'] + " " + bemenet, max_length=100, num_return_sequences=2)[1]['generated_text']
13
 
 
14
 
15
  interface = gr.Interface(fn=Bemenet,
16
  title="GPT-2",
17
- description="A GPT-2 egy előtanított nyelvi modell, ami nem párbeszédre lett optimalizálva. Használatakor figyelj erre, érdemesebb olyan bemenetekkel próbálkozni ami az eredeti tanítási módot használják ki.",
18
- inputs=gr.Textbox(),
19
- outputs=gr.Textbox())
20
 
21
  interface.launch()
 
11
  def Bemenet(bemenet):
12
  return generator(os.environ['SECRET_PROMPT'] + " " + bemenet, max_length=100, num_return_sequences=2)[1]['generated_text']
13
 
14
+ textbox = gr.Textbox()
15
 
16
  interface = gr.Interface(fn=Bemenet,
17
  title="GPT-2",
18
+ description="A GPT-2 egy előtanított nyelvi modell, ami nem párbeszédre lett optimalizálva. Használatakor figyelj erre, érdemesebb olyan bemenetekkel próbálkozni ami az eredeti tanítási célt használják ki.",
19
+ inputs=textbox,
20
+ outputs=textbox)
21
 
22
  interface.launch()