fix bugs

app.py

@@ -1,5 +1,5 @@
 import torch
-from NeuroPredPLM.predict import predict, batch_predict, batch_predict_beta
+from NeuroPredPLM.predict import predict, batch_predict
 import gradio as gr
 from io import StringIO
 from Bio import SeqIO
@@ -20,7 +20,7 @@ def batch_classifier(file, cutoff):
     for record in SeqIO.parse(file.name, 'fasta'):
         data.append((record.id, str(record.seq)))
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    neuropeptide_pred = batch_predict_beta(data, cutoff, '/mnt/d/protein-net/NeuroPred-PLM/NeuroPred-PLM/model.pth', device)
+    neuropeptide_pred = batch_predict(data, cutoff, '/mnt/d/protein-net/NeuroPred-PLM/NeuroPred-PLM/model.pth', device)
     return neuropeptide_pred
 
 with gr.Blocks() as demo: