Update app.py

app.py

@@ -134,10 +134,11 @@ def greet(Amino_Acid_Sequence):
     lstm_fam_id = prot_mappings['fam_asc2fam_id'][lstm_fam_asc]
     fam_asc = prot_mappings['id2fam_asc'][str(idx)]
     fam_id = prot_mappings['fam_asc2fam_id'][fam_asc]
-    idprobs = {prot_mappings['id2fam_asc'][str(i)]: float(joined_prediction[i]) for i in range(len(joined_prediction))}
-    famprobs = {prot_mappings['fam_asc2fam_id'][fam_asc]: pred  for fam_asc, pred in idprobs.items()}
+    joined_probs = {prot_mappings['id2fam_asc'][str(i)] + ' ' + prot_mappings['fam_asc2fam_id'][prot_mappings['id2fam_asc'][str(i)]]: float(joined_prediction[i]) for i in range(len(joined_prediction))}
+    cnn_probs = {prot_mappings['id2fam_asc'][str(i)] + ' ' + prot_mappings['fam_asc2fam_id'][prot_mappings['id2fam_asc'][str(i)]]: float(cnn_raw_prediction[i]) for i in range(len(cnn_raw_prediction))}
+    lstm_probs = {prot_mappings['id2fam_asc'][str(i)] + ' ' + prot_mappings['fam_asc2fam_id'][prot_mappings['id2fam_asc'][str(i)]]: float(lstm_raw_prediction[i]) for i in range(len(lstm_raw_prediction))}
     gc.collect()
-    return famprobs, idprobs, f"""
+    return joined_probs, cnn_probs, lstm_probs, f"""
 Input is {Amino_Acid_Sequence}.
 Processed input is:
 {processed_seq}
@@ -152,5 +153,5 @@ Raw Joined Prediction:
 {joined_prediction}
 """
 
-iface = gr.Interface(fn=greet, inputs="text", outputs=[gr.Label(num_top_classes=11, label="Family Accession Predictions"), gr.Label(num_top_classes=11, label="Family ID Predictions"), "text"])
+iface = gr.Interface(fn=greet, inputs="text", outputs=[gr.Label(num_top_classes=5, label="Ensemble Family Predictions"), gr.Label(num_top_classes=5, label="CNN Family Predictions"), gr.Label(num_top_classes=5, label="LSTM Family Predictions"), "text"])
 iface.launch()