Changed the labels to english

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from Model import LeNet
 
-labels = ['Zero','Um','Dois','Três','Quatro','Cinco','Seis','Sete','Oito', 'Nove']
+labels = ['Zero','One','Two','Three','Four','Five','Six','Seven','Eight', 'Nine']
 
 # Locate device
 if torch.cuda.is_available():
@@ -12,8 +12,6 @@ else:
     device = torch.device("cpu")
     print("CPU")
 
-
-
 # Loading model
 model = LeNet().to(device)
 model.load_state_dict(torch.load("model_mnist.pth", map_location=torch.device('cpu')))
@@ -28,6 +26,6 @@ def predict(input):
     confidences = {labels[i]: float(prediction[i]) for i in range(10)}    
   return confidences
 
-gr.Interface(title='Classificador de dígitos', fn=predict, 
+gr.Interface(title='Digit classifier', fn=predict, 
              inputs="sketchpad",
              outputs=gr.Label(num_top_classes=3)).launch(share=False, debug=True)