Update app.py

app.py

@@ -53,8 +53,8 @@ model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",f
 multitask_model.load_state_dict(torch.load(
    model_link #"multitask_model_3ep/pytorch_model.bin"
     ))
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-multitask_model.to(device)
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# multitask_model.to(device)
 predictions = []
 for _, batch in enumerate(features_dict["document"]['test']):
     for key, value in batch.items():
@@ -81,7 +81,7 @@ trainer = MultitaskTrainer(
         do_eval=True,
         # evaluation_strategy ="steps",
         # num_train_epochs=epochs,
-        fp16=True,
+        # fp16=True,
         # Adjust batch size if this doesn't fit on the Colab GPU
         per_device_train_batch_size=batch_size,
         per_device_eval_batch_size=batch_size,