joechou commited on
Commit
5e6557c
·
1 Parent(s): 6a5d569

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -178,6 +178,8 @@ cond = aspect+" "+cond
178
  #print(title)
179
  #print(fact)
180
  #print(cond)
 
 
181
  tokenizer = text_tokenizer
182
  title_token_ids=tokenizer.encode(title, add_special_tokens=False)
183
  condition_token_ids=cond_tokenizer.EncodeAsIds(cond)
@@ -193,7 +195,7 @@ dm = KobeDataModule(
193
  1,
194
  1,
195
  )
196
- '''
197
  for d in dm.test_dataloader():
198
  st.write("result:")
199
  st.write(''.join(model.test_step(d ,1)).replace(" ",""))
 
178
  #print(title)
179
  #print(fact)
180
  #print(cond)
181
+
182
+ '''
183
  tokenizer = text_tokenizer
184
  title_token_ids=tokenizer.encode(title, add_special_tokens=False)
185
  condition_token_ids=cond_tokenizer.EncodeAsIds(cond)
 
195
  1,
196
  1,
197
  )
198
+
199
  for d in dm.test_dataloader():
200
  st.write("result:")
201
  st.write(''.join(model.test_step(d ,1)).replace(" ",""))