Spaces:
Running
Running
meghanaraok
commited on
Commit
•
538f691
1
Parent(s):
e8d5255
Update app.py
Browse files
app.py
CHANGED
@@ -3,11 +3,11 @@ from torch import nn
|
|
3 |
import pandas as pd
|
4 |
import numpy as np
|
5 |
import gradio as gr
|
6 |
-
from LongLAT.
|
7 |
-
from HiLATmain.
|
8 |
-
from LongHiLATmain.
|
9 |
from run_coding import ModelArguments, DataTrainingArguments
|
10 |
-
from
|
11 |
from transformers import (
|
12 |
AutoConfig,
|
13 |
AutoTokenizer,
|
|
|
3 |
import pandas as pd
|
4 |
import numpy as np
|
5 |
import gradio as gr
|
6 |
+
from LongLAT.LongLAT.models.modeling import CodingModel as LongLATCodingModel, CodingModelConfig as LongLATCodingModelConfig
|
7 |
+
from LongLAT.HiLATmain.models.modeling import CodingModel, CodingModelConfig
|
8 |
+
from LongLAT.LongHiLATmain.models.modeling import CodingModel as LongHiLATCodingModel, CodingModelConfig as LongHiLATCodingModelConfig
|
9 |
from run_coding import ModelArguments, DataTrainingArguments
|
10 |
+
from LongLAT.LongLAT.models.utils import segment_tokenize_dataset
|
11 |
from transformers import (
|
12 |
AutoConfig,
|
13 |
AutoTokenizer,
|