Update app.py

app.py

@@ -20,21 +20,23 @@ ds = load_dataset(OUTPUTS_DATASET, split="train", use_auth_token=HF_TOKEN)
 def process():
     sample_ds = ds.shuffle().select(range(1))
     sample = sample_ds[0]
-    
+    prompt = sample["prompt"]
     
     df = pd.DataFrame.from_records(sample["filtered_outputs"])
         
-    input_text = "POST: "+ sample["prompt"]+ "\n\n RESPONSE A: Lime juice, and zest, then freeze in small quantities.\n\n RESPONSE B: Lime marmalade lol\n\n Which response is better? RESPONSE"
+    input_text = "POST: "+ prompt+ "\n\n RESPONSE A: Lime juice, and zest, then freeze in small quantities.\n\n RESPONSE B: Lime marmalade lol\n\n Which response is better? RESPONSE"
     x = tokenizer([input_text], return_tensors='pt').input_ids.to(device)
     y = model.generate(x, max_new_tokens=1)
     prefered = tokenizer.batch_decode(y, skip_special_tokens=True)[0] 
-    return sample["filtered_outputs"]
+    return prefered, df
 
 title = "Compare Instruction Models to see which one is more helpful"
 interface = gr.Interface(fn=process, 
                      inputs=[],
                      outputs=[
-                              gr.Textbox(label = "Responses")
+                              gr.Textbox(label = "Preferred Option"),
+                              gr.DataFrame(label = "Model Responses")
+                                 
                               ],
                      title=title,
                      )