Spaces:
Running
on
Zero
Running
on
Zero
Update model/model_manager.py
Browse files- 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 |
-
|
192 |
-
|
193 |
-
else:
|
194 |
-
|
195 |
return result
|
196 |
|
197 |
def generate_video_vg_api(self, prompt, model_name):
|
198 |
-
if self.NSFW_filter(prompt) == 'safe':
|
199 |
-
|
200 |
-
|
201 |
-
else:
|
202 |
-
|
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):
|