Update gui.py

gui.py

@@ -1,35 +1,21 @@
-from api_prediction import AptaTransPipeline_Dist
+from api_prediction import AptaBLE_Pipeline_Dist
 import gradio as gr
 import pandas as pd
 import torch
 import tempfile
 from tabulate import tabulate
-from PIL import Image
 import itertools
 import os
-import RNA
-import matplotlib.pyplot as plt
-import matplotlib.image as mpimg
 import random
-from scipy.cluster.hierarchy import dendrogram, linkage
 # Visualization
-from Bio.Phylo.PhyloXML import Phylogeny
-from Bio import SeqIO
-from Bio.Seq import Seq
-from Bio.SeqRecord import SeqRecord
-from Bio import AlignIO
-from Bio.Align.Applications import MafftCommandline
-from Bio import Phylo
-from Bio.Phylo.TreeConstruction import DistanceCalculator, DistanceTreeConstructor
-import io
 
 os.environ['GRADIO_SERVER_NAME'] = '0.0.0.0'
-title='DNAptaESM2 Model Infernence'
+title='DNAptaBLE Model Inference'
 desc='AptaBLE (cross-attention network), trained to predict the likelihood a DNA aptamer will form a complex with a target protein!\n\nPass in a FASTA-formatted file of all aptamers and input your protein target amino acid sequence. Your output scores are available for download via an Excel file.'
 
 global pipeline
 
-pipeline = AptaTransPipeline_Dist(
+pipeline = AptaBLE_Pipeline_Dist(
     lr=1e-6,
     weight_decay=None,
     epochs=None,