RamAnanth1 commited on
Commit
d7bea84
·
1 Parent(s): 82f1eb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,14 +41,14 @@ def process(model_A, model_B):
41
  x = tokenizer([input_text], return_tensors='pt').input_ids.to(device)
42
  y = model.generate(x, max_new_tokens=1)
43
  prefered = tokenizer.batch_decode(y, skip_special_tokens=True)[0]
44
- return df[df['model'].isin([model_A, model_B])], prefered
45
 
46
  title = "Compare Instruction Models to see which one is more helpful"
47
  interface = gr.Interface(fn=process,
48
  inputs=[gr.Dropdown(choices=model_list, value=model_list[0], label='Model A'),
49
  gr.Dropdown(choices=model_list, value=model_list[1], label='Model B')],
50
  outputs=[
51
-
52
  gr.DataFrame(label = "Model Responses"),
53
  gr.Textbox(label = "Preferred Option"),
54
 
 
41
  x = tokenizer([input_text], return_tensors='pt').input_ids.to(device)
42
  y = model.generate(x, max_new_tokens=1)
43
  prefered = tokenizer.batch_decode(y, skip_special_tokens=True)[0]
44
+ return prompt,df[df['model'].isin([model_A, model_B])], prefered
45
 
46
  title = "Compare Instruction Models to see which one is more helpful"
47
  interface = gr.Interface(fn=process,
48
  inputs=[gr.Dropdown(choices=model_list, value=model_list[0], label='Model A'),
49
  gr.Dropdown(choices=model_list, value=model_list[1], label='Model B')],
50
  outputs=[
51
+ gr.Textbox("Context")
52
  gr.DataFrame(label = "Model Responses"),
53
  gr.Textbox(label = "Preferred Option"),
54