Updated

.ipynb_checkpoints/app-checkpoint.py

@@ -584,19 +584,7 @@ with gr.Blocks(css="""
         else:
             return gr.update(visible=False), gr.update(visible=True)
 
-    def predict_utils(sequence):
-        input_ids = tokenizer(" ".join(sequence), return_tensors="pt").input_ids.to(device)
-        with torch.no_grad():
-            outputs = model(input_ids).logits.detach().cpu().numpy().squeeze()
-    
-        # Calculate scores and normalize them
-        raw_scores = expit(outputs[:, 1] - outputs[:, 0])
-        normalized_scores = normalize_scores(raw_scores)
     
-        return {
-        "raw_scores": raw_scores.tolist(),
-        "normalized_scores": normalized_scores.tolist()
-    }
         
     mode.change(
         toggle_mode,
@@ -636,6 +624,25 @@ with gr.Blocks(css="""
         outputs=[predictions_output, molecule_output, download_output]
     )
 
+    def predict_utils(sequence):
+        input_ids = tokenizer(" ".join(sequence), return_tensors="pt").input_ids.to(device)
+        with torch.no_grad():
+            outputs = model(input_ids).logits.detach().cpu().numpy().squeeze()
+    
+        # Calculate scores and normalize them
+        raw_scores = expit(outputs[:, 1] - outputs[:, 0])
+        normalized_scores = normalize_scores(raw_scores)
+    
+        return {
+        "raw_scores": raw_scores.tolist(),
+        "normalized_scores": normalized_scores.tolist()
+    }
+
+    gr.Interface(
+        fn=predict_utils,
+        inputs=gr.Textbox(visible=False),
+        outputs=gr.Textbox(visible=False)
+    )
+
     
-demo.load(predict_utils, inputs=gr.Textbox(), outputs=gr.Textbox())
 demo.launch(share=True)

app.py

@@ -584,19 +584,7 @@ with gr.Blocks(css="""
         else:
             return gr.update(visible=False), gr.update(visible=True)
 
-    def predict_utils(sequence):
-        input_ids = tokenizer(" ".join(sequence), return_tensors="pt").input_ids.to(device)
-        with torch.no_grad():
-            outputs = model(input_ids).logits.detach().cpu().numpy().squeeze()
-    
-        # Calculate scores and normalize them
-        raw_scores = expit(outputs[:, 1] - outputs[:, 0])
-        normalized_scores = normalize_scores(raw_scores)
     
-        return {
-        "raw_scores": raw_scores.tolist(),
-        "normalized_scores": normalized_scores.tolist()
-    }
         
     mode.change(
         toggle_mode,
@@ -636,6 +624,25 @@ with gr.Blocks(css="""
         outputs=[predictions_output, molecule_output, download_output]
     )
 
+    def predict_utils(sequence):
+        input_ids = tokenizer(" ".join(sequence), return_tensors="pt").input_ids.to(device)
+        with torch.no_grad():
+            outputs = model(input_ids).logits.detach().cpu().numpy().squeeze()
+    
+        # Calculate scores and normalize them
+        raw_scores = expit(outputs[:, 1] - outputs[:, 0])
+        normalized_scores = normalize_scores(raw_scores)
+    
+        return {
+        "raw_scores": raw_scores.tolist(),
+        "normalized_scores": normalized_scores.tolist()
+    }
+
+    gr.Interface(
+        fn=predict_utils,
+        inputs=gr.Textbox(visible=False),
+        outputs=gr.Textbox(visible=False)
+    )
+
     
-demo.load(predict_utils, inputs=gr.Textbox(), outputs=gr.Textbox())
 demo.launch(share=True)