Hugging Face's logo

Spaces:
igashov
/
DiffLinker
Runtime error

App Files Community
3
igashov commited on
Commit
98807df
·
1 Parent(s): 5de53c3

custom ip

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from src.lightning import DDPM
15
  from src.linker_size_lightning import SizeClassifier
16
 
17
  parser = argparse.ArgumentParser()
18
- parser.add_argument(name='ip', type=str, default=None)
19
  args = parser.parse_args()
20
 
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
15
  from src.linker_size_lightning import SizeClassifier
16
 
17
  parser = argparse.ArgumentParser()
18
+ parser.add_argument('ip', type=str, default=None)
19
  args = parser.parse_args()
20
 
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")