RamAnanth1 commited on
Commit
a2f31b8
·
1 Parent(s): dfc8b82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def process(model_A, model_B):
37
  x = tokenizer([input_text], return_tensors='pt').input_ids.to(device)
38
  y = model.generate(x, max_new_tokens=1)
39
  prefered = tokenizer.batch_decode(y, skip_special_tokens=True)[0]
40
- return pd.concat([response_A,response_B], prefered
41
 
42
  title = "Compare Instruction Models to see which one is more helpful"
43
  interface = gr.Interface(fn=process,
 
37
  x = tokenizer([input_text], return_tensors='pt').input_ids.to(device)
38
  y = model.generate(x, max_new_tokens=1)
39
  prefered = tokenizer.batch_decode(y, skip_special_tokens=True)[0]
40
+ return pd.concat([response_A,response_B]), prefered
41
 
42
  title = "Compare Instruction Models to see which one is more helpful"
43
  interface = gr.Interface(fn=process,