Spaces:
Runtime error
Runtime error
fix import
Browse files- CI/test_ci.py +1 -0
- src/modello.py +1 -0
CI/test_ci.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import torch
|
| 2 |
from src.modello import Modello
|
| 3 |
from src.dataset import LoadDataset
|
|
|
|
| 1 |
+
# Utilities
|
| 2 |
import torch
|
| 3 |
from src.modello import Modello
|
| 4 |
from src.dataset import LoadDataset
|
src/modello.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from transformers import pipeline
|
| 2 |
|
| 3 |
class Modello :
|
|
|
|
| 1 |
+
import torch
|
| 2 |
from transformers import pipeline
|
| 3 |
|
| 4 |
class Modello :
|