acmyu commited on
Commit
95510cb
·
1 Parent(s): ce2fb73

fix eval api bugs

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. evaluate.py +7 -3
app.py CHANGED
@@ -70,7 +70,7 @@ with gr.Blocks() as demo:
70
  )
71
 
72
 
73
- demo.launch(share=True)
74
 
75
 
76
 
 
70
  )
71
 
72
 
73
+ demo.launch(share=True, show_error=True)
74
 
75
 
76
 
evaluate.py CHANGED
@@ -113,9 +113,13 @@ def compute_fid(img1, img2):
113
 
114
  def get_score(item, image_paths, video_path, train_steps=100, inference_steps=10, fps=12, bg_remove=False):
115
  images = []
116
- for path in image_paths:
117
- img = Image.open(path)
118
- images.append([img])
 
 
 
 
119
 
120
  results = []
121
  results_base = []
 
113
 
114
  def get_score(item, image_paths, video_path, train_steps=100, inference_steps=10, fps=12, bg_remove=False):
115
  images = []
116
+ if not isinstance(image_paths[0], str):
117
+ images = image_paths
118
+ else:
119
+ for path in image_paths:
120
+ print(path)
121
+ img = Image.open(path)
122
+ images.append([img])
123
 
124
  results = []
125
  results_base = []