Spaces:

telecomadm1145
/

ImageRestore

Sleeping

telecomadm1145 commited on 26 days ago

Commit

f7d97be

verified ·

1 Parent(s): 1231a49

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -831,7 +831,7 @@ def create_demo(restorer: ImageRestorer):
 # 主函数
 # -----------------------------------------------------------------------------
 def main():
-    restorer = ImageRestorer()
     create_demo(restorer).queue(
         max_size=10,
         default_concurrency_limit=2

 # 主函数
 # -----------------------------------------------------------------------------
 def main():
+    restorer = ImageRestorer(model_path=MODEL_PATH,repo_id=MODEL_REPO_ID)
     create_demo(restorer).queue(
         max_size=10,
         default_concurrency_limit=2