meghanaraok commited on
Commit
538f691
1 Parent(s): e8d5255

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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.hilat.models.modeling import CodingModel as LongLATCodingModel, CodingModelConfig as LongLATCodingModelConfig
7
- from HiLATmain.hilat.models.modeling import CodingModel, CodingModelConfig
8
- from LongHiLATmain.hilat.models.modeling import CodingModel as LongHiLATCodingModel, CodingModelConfig as LongHiLATCodingModelConfig
9
  from run_coding import ModelArguments, DataTrainingArguments
10
- from HiLATmain.hilat.models.utils import segment_tokenize_dataset
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,