Spaces:
Build error
Build error
Commit
·
a9e3b02
1
Parent(s):
6e254e5
Update chat.py
Browse files
chat.py
CHANGED
|
@@ -168,21 +168,21 @@ class Chat:
|
|
| 168 |
"text": server_error_msg,
|
| 169 |
"error_code": 1,
|
| 170 |
}
|
| 171 |
-
yield json.dumps(ret).encode()
|
| 172 |
except torch.cuda.CudaError as e:
|
| 173 |
print("Caught torch.cuda.CudaError:", e)
|
| 174 |
ret = {
|
| 175 |
"text": server_error_msg,
|
| 176 |
"error_code": 1,
|
| 177 |
}
|
| 178 |
-
yield json.dumps(ret).encode()
|
| 179 |
except Exception as e:
|
| 180 |
print("Caught Unknown Error", e)
|
| 181 |
ret = {
|
| 182 |
"text": server_error_msg,
|
| 183 |
"error_code": 1,
|
| 184 |
}
|
| 185 |
-
yield json.dumps(ret).encode()
|
| 186 |
|
| 187 |
|
| 188 |
if __name__ == "__main__":
|
|
|
|
| 168 |
"text": server_error_msg,
|
| 169 |
"error_code": 1,
|
| 170 |
}
|
| 171 |
+
yield json.dumps(ret).encode()
|
| 172 |
except torch.cuda.CudaError as e:
|
| 173 |
print("Caught torch.cuda.CudaError:", e)
|
| 174 |
ret = {
|
| 175 |
"text": server_error_msg,
|
| 176 |
"error_code": 1,
|
| 177 |
}
|
| 178 |
+
yield json.dumps(ret).encode()
|
| 179 |
except Exception as e:
|
| 180 |
print("Caught Unknown Error", e)
|
| 181 |
ret = {
|
| 182 |
"text": server_error_msg,
|
| 183 |
"error_code": 1,
|
| 184 |
}
|
| 185 |
+
yield json.dumps(ret).encode()
|
| 186 |
|
| 187 |
|
| 188 |
if __name__ == "__main__":
|