ChaosTong commited on
Commit
ef81edb
·
1 Parent(s): 52f1ceb
Files changed (2) hide show
  1. app.py +2 -1
  2. inference.py +3 -2
app.py CHANGED
@@ -87,4 +87,5 @@ gr.Interface(
87
  ['example/landscape/pexels-camilacarneiro-6318793.jpg', 'AnimeGANv2_Hayao', None],
88
  ['example/landscape/pexels-nandhukumar-450441.jpg', 'AnimeGANv2_Hayao', None],
89
  ]
90
- ).launch(server_name="0.0.0.0")
 
 
87
  ['example/landscape/pexels-camilacarneiro-6318793.jpg', 'AnimeGANv2_Hayao', None],
88
  ['example/landscape/pexels-nandhukumar-450441.jpg', 'AnimeGANv2_Hayao', None],
89
  ]
90
+ ).launch()
91
+ # server_name="0.0.0.0", server_port=8080
inference.py CHANGED
@@ -43,8 +43,9 @@ def auto_load_weight(weight, version=None, map_location=None):
43
  """Auto load Generator version from weight."""
44
  project_dir = os.path.dirname(os.path.abspath(__file__))
45
  cache_dir = os.path.join(project_dir, ".cache")
46
- weight_name = os.path.basename(weight).lower()
47
  cached_weight = os.path.join(cache_dir, weight_name)
 
48
 
49
  # Check if the cached weight file exists
50
  if os.path.exists(cached_weight):
@@ -67,7 +68,7 @@ def auto_load_weight(weight, version=None, map_location=None):
67
  version = RELEASED_WEIGHTS[weight_name][0]
68
  return auto_load_weight(weight, version=version, map_location=map_location)
69
 
70
- elif weight_name.startswith("generatorv2"):
71
  cls = GeneratorV2
72
  elif weight_name.startswith("generatorv3"):
73
  cls = GeneratorV3
 
43
  """Auto load Generator version from weight."""
44
  project_dir = os.path.dirname(os.path.abspath(__file__))
45
  cache_dir = os.path.join(project_dir, ".cache")
46
+ weight_name = os.path.basename(weight)#.lower()
47
  cached_weight = os.path.join(cache_dir, weight_name)
48
+ print(project_dir, cache_dir, weight, weight_name, cached_weight)
49
 
50
  # Check if the cached weight file exists
51
  if os.path.exists(cached_weight):
 
68
  version = RELEASED_WEIGHTS[weight_name][0]
69
  return auto_load_weight(weight, version=version, map_location=map_location)
70
 
71
+ elif weight_name.startswith("GeneratorV2"):
72
  cls = GeneratorV2
73
  elif weight_name.startswith("generatorv3"):
74
  cls = GeneratorV3