Spaces:
Runtime error
Runtime error
Commit
·
d7bea84
1
Parent(s):
82f1eb9
Update app.py
Browse files
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 |
|