Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -75,13 +75,19 @@ if token:
|
|
| 75 |
else:
|
| 76 |
print("讗讝讛专讛: 讟讜拽谉 HF_TOKEN 诇讗 诪讜讙讚专 讘住讘讬讘转 讛注讘讜讚讛. 讬讬转讻谉 砖诇讗 转讛讬讛 讙讬砖讛 诇诪讜讚诇讬诐 诪讜讙讘诇讬诐.")
|
| 77 |
|
| 78 |
-
#
|
| 79 |
default_model_name = "facebook/opt-125m"
|
| 80 |
current_model_name = default_model_name
|
| 81 |
-
|
| 82 |
-
# 诪砖转谞讛 讙诇讜讘诇讬 诇讙谞专讟讜专
|
| 83 |
generator = None
|
| 84 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
def load_model(model_name, status_box=None):
|
| 86 |
"""讟注讬谞转 诪讜讚诇 讜注讚讻讜谉 住讟讟讜住"""
|
| 87 |
global generator, current_model_name
|
|
|
|
| 75 |
else:
|
| 76 |
print("讗讝讛专讛: 讟讜拽谉 HF_TOKEN 诇讗 诪讜讙讚专 讘住讘讬讘转 讛注讘讜讚讛. 讬讬转讻谉 砖诇讗 转讛讬讛 讙讬砖讛 诇诪讜讚诇讬诐 诪讜讙讘诇讬诐.")
|
| 77 |
|
| 78 |
+
# 讛讙讚专转 诪砖转谞讬诐 讙诇讜讘诇讬讬诐
|
| 79 |
default_model_name = "facebook/opt-125m"
|
| 80 |
current_model_name = default_model_name
|
|
|
|
|
|
|
| 81 |
generator = None
|
| 82 |
|
| 83 |
+
# 诪讙讚讬专 讗转 generator 讻诪砖转谞讛 讙诇讜讘诇讬 讘讻诇 讛驻讜谞拽爪讬讜转
|
| 84 |
+
def init_globals():
|
| 85 |
+
global generator, current_model_name
|
| 86 |
+
# 砖诐 诪砖转谞讬诐 讙诇讜讘诇讬讬诐 诪讗讜转讞诇讬诐
|
| 87 |
+
|
| 88 |
+
# 诪讗转讞诇 讗转 讛诪砖转谞讬诐 讛讙诇讜讘诇讬讬诐
|
| 89 |
+
init_globals()
|
| 90 |
+
|
| 91 |
def load_model(model_name, status_box=None):
|
| 92 |
"""讟注讬谞转 诪讜讚诇 讜注讚讻讜谉 住讟讟讜住"""
|
| 93 |
global generator, current_model_name
|