Update app.py

app.py

@@ -112,8 +112,7 @@ def compute_metrics(p: EvalPrediction):
         labels=p.label_ids)
     return result
 
-
-trainer = Trainer(
+"""trainer = Trainer(
     model,
     args,
     train_dataset=encoded_dataset["train"],
@@ -121,14 +120,22 @@ trainer = Trainer(
     tokenizer=tokenizer,
     compute_metrics=compute_metrics
 )
-
+"""
 
 # REMOVE THIS IN COLAB #############
 
-title = 'Text market sentiment'
 text_ = "Bitcoin to the moon"
 model = torch.load("./model.pt", map_location=torch.device('cpu'))
 
+trainer = Trainer(
+    model,
+    args,
+    train_dataset=encoded_dataset["train"],
+    eval_dataset=encoded_dataset["validation"],
+    tokenizer=tokenizer,
+    compute_metrics=compute_metrics
+)
+
 tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 
 def predict(text):