elucidator8918 commited on
Commit
530d702
·
verified ·
1 Parent(s): 13fd1be

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -0
tasks/text.py CHANGED
@@ -1,5 +1,6 @@
1
  import torch
2
  import random
 
3
  from fastapi import APIRouter
4
  from datetime import datetime
5
  from datasets import load_dataset
 
1
  import torch
2
  import random
3
+ from torch import nn
4
  from fastapi import APIRouter
5
  from datetime import datetime
6
  from datasets import load_dataset