Update app.py

app.py

@@ -106,7 +106,7 @@ def greet(Amino_Acid_Sequence):
     processed_seq = Sequence.prepare(Amino_Acid_Sequence)
     raw_prediction = model.predict(processed_seq)[0]
     idx = raw_prediction.argmax()
-    fam_asc = prot_mappings['id2fam_asc'][idx]
+    fam_asc = prot_mappings['id2fam_asc'][str(idx)]
     fam_id = prot_mappings['fam_asc2fam_id'][fam_asc]
     gc.collect()
     return f"Input is {Amino_Acid_Sequence}.\nProcessed input is:\n{processed_seq}\n\nModel makes prediction:\n{raw_prediction}\nWhich gives index {idx}.\nWhich maps to family ascension {fam_asc} and ID {fam_id}"