Spaces:

nateraw
/

custom-pipeline-server

Sleeping

nateraw commited on Apr 12, 2023

Commit

57e5ee1

1 Parent(s): 37cbaed

Synced repo using 'sync_with_huggingface' Github Action

Files changed (1) hide show

main.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import time
 from closedai import ClosedAIPipeline
-from closedai.server import app, data  # noqa
 class MyPipeline(ClosedAIPipeline):
@@ -19,5 +19,4 @@ class MyPipeline(ClosedAIPipeline):
             time.sleep(1)
-pipeline = MyPipeline()
-data["pipeline"] = pipeline

 import time
 from closedai import ClosedAIPipeline
+from closedai.server import app, register_model  # noqa
 class MyPipeline(ClosedAIPipeline):
             time.sleep(1)
+register_model("my_model", MyPipeline())