Spaces:

AtomBiow
/

AptaBLE

Sleeping

Atom Bioworks commited on Sep 18

Commit

529b593

•

1 Parent(s): b200805

Update gui.py

Files changed (1) hide show

gui.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from api_prediction import AptaBLE_Pipeline_Dist
 import gradio as gr
 import pandas as pd
 import torch
@@ -15,7 +15,7 @@ desc='AptaBLE (cross-attention network), trained to predict the likelihood a DNA
 global pipeline
-pipeline = AptaBLE_Pipeline_Dist(
     lr=1e-6,
     weight_decay=None,
     epochs=None,

+from api_prediction import AptaBLE_Pipeline
 import gradio as gr
 import pandas as pd
 import torch
 global pipeline
+pipeline = AptaBLE_Pipeline(
     lr=1e-6,
     weight_decay=None,
     epochs=None,