Update app.py

app.py

@@ -46,7 +46,7 @@ device = torch.device('cpu')
 
 # Load the saved model and map it to the CPU
 loaded_model = BertForSequenceClassification.from_pretrained(model_name, num_labels=2)
-loaded_model.load_state_dict(torch.load('/content/drive/MyDrive/best_model8.pt', map_location=device))
+loaded_model.load_state_dict(torch.load('best_model8.pt', map_location=device))
 loaded_model.to(device)
 
 # Define the Gradio interface
@@ -58,7 +58,7 @@ iface = gr.Interface(
         ["Seu Comunista!"],
         ['Os imigrantes não deveriam ser impedidos de entrar no meu país'],
         ['Os imigrantes deveriam ser impedidos de entrar no meu país'],
-        ['eu te amo],
+        ['eu te amo'],
         ['aquele cara é um babaca'],
     ]
 )