Update app.py

app.py

@@ -1,4 +1,4 @@
-"""
+'''
 import argparse
 import random
 import numpy as np
@@ -31,7 +31,7 @@ args.vocab_file= "bert-base-chinese"
 args.cond_vocab_file = "./vocab.cond.model"
 add_args(args)
  
-
+'''
     
 #model = KobeModel(args)
 
@@ -60,14 +60,14 @@ cond_tokenizer.Load(args.cond_vocab_file)
 #model = model.load_from_checkpoint("/root/kobe-v2/37ht1cvz/checkpoints/epoch=11-step=396384.ckpt", args=args)
 
 trainer = pl.Trainer(accelerator='gpu', devices=1, max_epochs=-1)
-"""
+
 
 
 import streamlit as st
 
 st.write("Most appearing words including stopwords")
 
-"""
+
 choice = st.selectbox(
 
     'Select the items you want?',
@@ -192,7 +192,8 @@ dm = KobeDataModule(
         1,
         1,
     )
+'''
 for d in dm.test_dataloader():
     st.write("result:")
     st.write(''.join(model.test_step(d ,1)).replace(" ",""))
-""" 
+'''