Update app.py

app.py

@@ -12,6 +12,11 @@ def inference(img, ver):
     return out
     
 def generate(text,guess):
+    if ver == "Hidden Identity 1":
+        guess = "Michael Jackson"
+    else:
+        guess = "Brad Pitt"
+        
     inputs =  tokenizer(f"Answer with yes or no the following question about {guess}: {text}?", return_tensors="pt")
     return model.generate(**inputs)[0]
 
@@ -23,7 +28,7 @@ examples = [
 demo = gr.Interface(
     fn=generate,
     inputs=[gr.inputs.Textbox(lines=5, label="Input Text"),
-    		gr.inputs.Radio("Hidden Identity 1", type="value", default='Hidden Identity 1', label='Hidden identity')],
+    		gr.inputs.Radio("Hidden Identity 1", type="text", default='Hidden Identity 1', label='Hidden identity')],
     outputs=gr.outputs.Textbox(label="Generated Text"),
     examples=examples
 )