Update app.py

app.py

@@ -42,7 +42,7 @@ def predict(inp):
     with torch.no_grad():
         prediction = torch.nn.functional.softmax(model(inp)[0], dim=0)
         #print(prediction)
-        confidences = {list(common_to_latin.keys())[i]: float(prediction[i]) for i in range(26)}
+        confidences = {list(common_to_latin.keys())[i]: float(prediction[i]) for i in range(32)}
     return confidences 
 
 interface = gr.Interface(fn=predict,