wenhu commited on
Commit
b3e524c
1 Parent(s): fcd973b

Update model/model_manager.py

Browse files
Files changed (1) hide show
  1. model/model_manager.py +10 -10
model/model_manager.py CHANGED
@@ -187,19 +187,19 @@ class ModelManager:
187
 
188
  @spaces.GPU(duration=150)
189
  def generate_video_vg(self, prompt, model_name):
190
- if self.NSFW_filter(prompt) == 'safe':
191
- pipe = self.load_model_pipe(model_name)
192
- result = pipe(prompt=prompt)
193
- else:
194
- result = ''
195
  return result
196
 
197
  def generate_video_vg_api(self, prompt, model_name):
198
- if self.NSFW_filter(prompt) == 'safe':
199
- pipe = self.load_model_pipe(model_name)
200
- result = pipe(prompt=prompt)
201
- else:
202
- result = ''
203
  return result
204
 
205
  def generate_video_vg_museum(self, model_name):
 
187
 
188
  @spaces.GPU(duration=150)
189
  def generate_video_vg(self, prompt, model_name):
190
+ # if self.NSFW_filter(prompt) == 'safe':
191
+ pipe = self.load_model_pipe(model_name)
192
+ result = pipe(prompt=prompt)
193
+ # else:
194
+ # result = ''
195
  return result
196
 
197
  def generate_video_vg_api(self, prompt, model_name):
198
+ # if self.NSFW_filter(prompt) == 'safe':
199
+ pipe = self.load_model_pipe(model_name)
200
+ result = pipe(prompt=prompt)
201
+ # else:
202
+ # result = ''
203
  return result
204
 
205
  def generate_video_vg_museum(self, model_name):