AlejandroLanaspa commited on
Commit
700abd4
·
1 Parent(s): d21cbfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -4,12 +4,6 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
  model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
5
  tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
6
 
7
- def inference(img, ver):
8
- if ver == "Hidden Identity 1":
9
- out = "Michael Jackson"
10
- else:
11
- out = "Brad Pitt"
12
- return out
13
 
14
  def generate(text,guess):
15
  if ver == "Hidden Identity 1":
@@ -28,7 +22,7 @@ examples = [
28
  demo = gr.Interface(
29
  fn=generate,
30
  inputs=[gr.inputs.Textbox(lines=5, label="Input Text"),
31
- gr.inputs.Radio("Hidden Identity 1", type="text", default='Hidden Identity 1', label='Hidden identity')],
32
  outputs=gr.outputs.Textbox(label="Generated Text"),
33
  examples=examples
34
  )
 
4
  model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
5
  tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
6
 
 
 
 
 
 
 
7
 
8
  def generate(text,guess):
9
  if ver == "Hidden Identity 1":
 
22
  demo = gr.Interface(
23
  fn=generate,
24
  inputs=[gr.inputs.Textbox(lines=5, label="Input Text"),
25
+ gr.inputs.Radio(["Hidden Identity 1","Hidden Identity 2"], type="value", default='Hidden Identity 1', label='Hidden identity')],
26
  outputs=gr.outputs.Textbox(label="Generated Text"),
27
  examples=examples
28
  )