Spaces:
Build error
Build error
app.py
CHANGED
|
@@ -130,7 +130,6 @@ if __name__ == '__main__':
|
|
| 130 |
args = parse_args(sys.argv[1:])
|
| 131 |
train = load_data_and_prepare(args.data_dir, args.excel_file, args.mode, args.scale, args.smote)
|
| 132 |
model = load_model(args.model_name_or_path)
|
| 133 |
-
device = torch.device(args.device)
|
| 134 |
examples = view.examples
|
| 135 |
description = view.description
|
| 136 |
title_markdown = view.title_markdown
|
|
|
|
| 130 |
args = parse_args(sys.argv[1:])
|
| 131 |
train = load_data_and_prepare(args.data_dir, args.excel_file, args.mode, args.scale, args.smote)
|
| 132 |
model = load_model(args.model_name_or_path)
|
|
|
|
| 133 |
examples = view.examples
|
| 134 |
description = view.description
|
| 135 |
title_markdown = view.title_markdown
|