Update Demo.py

Demo.py

@@ -548,8 +548,9 @@ language_map = {
     'zza': "Zaza"
   }
 
-language = language_map[output['language'][0].result]
-confidence = round(float(output['language'][0].metadata[language])*100, 2)
+abbreviation = output['language'][0].result
+language = language_map[abbreviation]
+confidence = round(float(output['language'][0].metadata[abbreviation])*100, 2)
 
-st.markdown(f"This text is in **{language} ({output['language'][0].result})** language.")
+st.markdown(f"This text is in **{language} ({abbreviation})** language.")
 st.markdown(f"Classification Confidence: **{confidence}%**")