RamAnanth1 commited on
Commit
32c4dc8
·
1 Parent(s): d552000

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ 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 pd.concat([response_A_df,response_B_df]), prefered
45
 
46
  title = "Compare Instruction Models to see which one is more helpful"
47
  interface = gr.Interface(fn=process,
 
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, prefered
45
 
46
  title = "Compare Instruction Models to see which one is more helpful"
47
  interface = gr.Interface(fn=process,