Update tasks/text.py

tasks/text.py

@@ -1,6 +1,6 @@
 import torch
 import random
-from torch import nn
+import torch.nn as nn
 from fastapi import APIRouter
 from datetime import datetime
 from datasets import load_dataset