Update app.py

app.py

@@ -41,14 +41,14 @@ def process(model_A, model_B):
     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  df[df['model'].isin([model_A, model_B])], prefered
+    return  prompt,df[df['model'].isin([model_A, model_B])], prefered
 
 title = "Compare Instruction Models to see which one is more helpful"
 interface = gr.Interface(fn=process, 
                      inputs=[gr.Dropdown(choices=model_list, value=model_list[0], label='Model A'),
                             gr.Dropdown(choices=model_list, value=model_list[1], label='Model B')],
                      outputs=[
-                              
+                              gr.Textbox("Context")
                               gr.DataFrame(label = "Model Responses"),
                               gr.Textbox(label = "Preferred Option"),