Update app.py

app.py

@@ -72,14 +72,12 @@ def predict(cfg, sequence):
         cfg.max_length += 1
 
     seed_everything(cfg.seed)
-    error
     df = pd.DataFrame({cfg.sequence_col: [sequence]})
 
     tokenizer = AutoTokenizer.from_pretrained(
         cfg.model_path, padding_side=cfg.padding_side
     )
     cfg.tokenizer = tokenizer
-
     dataset = PLTNUMDataset(cfg, df, train=False)
     dataloader = DataLoader(
         dataset,
@@ -93,7 +91,6 @@ def predict(cfg, sequence):
     model = PLTNUM_PreTrainedModel.from_pretrained(cfg.model_path, cfg=cfg)
     model.to(cfg.device)
 
-    # predictions = predict_fn(loader, model, cfg)
     model.eval()
     predictions = []
 
@@ -110,7 +107,7 @@ def predict(cfg, sequence):
     outputs = {}
     outputs["raw prediction values"] = predictions
     outputs["binary prediction values"] = [1 if x > 0.5 else 0 for x in predictions]
-    return outputs
+    return str(outputs)
 
 
 # Gradio Interface