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):
|