Spaces:
Runtime error
Runtime error
New Updates
Browse files- src/models/model.py +4 -2
- src/models/predict_model.py +1 -1
src/models/model.py
CHANGED
@@ -2,7 +2,7 @@ import time
|
|
2 |
import torch
|
3 |
import numpy as np
|
4 |
import pandas as pd
|
5 |
-
from
|
6 |
from transformers import (
|
7 |
AdamW,
|
8 |
T5ForConditionalGeneration,
|
@@ -302,7 +302,9 @@ class Summarization:
|
|
302 |
tokenizer=self.tokenizer, model=self.model, output=outputdir
|
303 |
)
|
304 |
|
305 |
-
logger = MLFlowLogger(experiment_name="Summarization",tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
|
|
|
|
|
306 |
|
307 |
early_stop_callback = (
|
308 |
[
|
|
|
2 |
import torch
|
3 |
import numpy as np
|
4 |
import pandas as pd
|
5 |
+
from dagshub.pytorch_lightning import DAGsHubLogger
|
6 |
from transformers import (
|
7 |
AdamW,
|
8 |
T5ForConditionalGeneration,
|
|
|
302 |
tokenizer=self.tokenizer, model=self.model, output=outputdir
|
303 |
)
|
304 |
|
305 |
+
#logger = MLFlowLogger(experiment_name="Summarization",tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
|
306 |
+
|
307 |
+
logger = DAGsHubLogger()
|
308 |
|
309 |
early_stop_callback = (
|
310 |
[
|
src/models/predict_model.py
CHANGED
@@ -1,5 +1,5 @@
|
|
|
|
1 |
from .model import Summarization
|
2 |
-
from .make_dataset import make_dataset
|
3 |
|
4 |
def predict_model(text):
|
5 |
"""
|
|
|
1 |
+
from src.data.make_dataset import make_dataset
|
2 |
from .model import Summarization
|
|
|
3 |
|
4 |
def predict_model(text):
|
5 |
"""
|