Spaces:
Running
on
Zero
Running
on
Zero
Update model/model_manager.py
Browse files- model/model_manager.py +2 -0
model/model_manager.py
CHANGED
|
@@ -66,6 +66,7 @@ class ModelManager:
|
|
| 66 |
pipe = self.load_model_pipe(model_name)
|
| 67 |
result = pipe(prompt=prompt)
|
| 68 |
else:
|
|
|
|
| 69 |
result = ''
|
| 70 |
return result
|
| 71 |
|
|
@@ -75,6 +76,7 @@ class ModelManager:
|
|
| 75 |
pipe = self.load_model_pipe(model_name)
|
| 76 |
result = pipe(prompt=prompt)
|
| 77 |
else:
|
|
|
|
| 78 |
result = ''
|
| 79 |
return result
|
| 80 |
|
|
|
|
| 66 |
pipe = self.load_model_pipe(model_name)
|
| 67 |
result = pipe(prompt=prompt)
|
| 68 |
else:
|
| 69 |
+
print(f'The prompt "{prompt}" is not safe')
|
| 70 |
result = ''
|
| 71 |
return result
|
| 72 |
|
|
|
|
| 76 |
pipe = self.load_model_pipe(model_name)
|
| 77 |
result = pipe(prompt=prompt)
|
| 78 |
else:
|
| 79 |
+
print(f'The prompt "{prompt}" is not safe')
|
| 80 |
result = ''
|
| 81 |
return result
|
| 82 |
|