Update app.py

app.py

@@ -22,7 +22,6 @@ class CFG():
     num_return_sequences = 5
     seed = 42
 
-print('ok')
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
@@ -35,8 +34,8 @@ def seed_everything(seed=42):
     torch.backends.cudnn.deterministic = True
 seed_everything(seed=CFG.seed) 
 
-print('ok')
-print(CFG.input_data)
+
+st.json(CFG.input_data)
 tokenizer = AutoTokenizer.from_pretrained(CFG.model_name_or_path, return_tensors='pt')
     
 if CFG.model == 't5':
@@ -62,5 +61,4 @@ if type(mol) == None:
 output += scores
 output = [input_compound] + output
 output_df = pd.DataFrame(np.array(output).reshape(1, -1), columns=['input'] + [f'{i}th' for i in range(CFG.num_beams)] + ['valid compound'] + [f'{i}th score' for i in range(CFG.num_beams)] + ['valid compound score'])
-print(output)
-print('ok')
+st.json(output)