feat: change result structure

app.py

@@ -136,6 +136,12 @@ def gradio_process_fasta(fasta_text):
     """
     fasta_dict = process_fasta(fasta_text)
     result = predictor.predict(fasta_dict)
+    new_result = {}
+    for key, value in result.items():
+        if value[1] == 1:
+            new_result[key] = {"probability": value[0], "class": "bitter"}
+        else:
+            new_result[key] = {"probability": 1 - value[0], "class": "non-bitter"}
     return result
 
 
@@ -145,7 +151,7 @@ interface = gr.Interface(
         label="Enter FASTA format text", lines=10, placeholder=">id1\nATGC\n>id2\nCGTA"
     ),
     outputs=gr.JSON(label="Processed FASTA Dictionary with Probabilities and Classes"),
-    title="FASTA to Dictionary with Probabilities and Classes",
+    title="xBitterT5",
     description=("Enter a FASTA-formatted text"),
 )
 # Launch the Gradio app