fix: stub to app

limcheekin · Jul 12, 2024 · a1eb778 · a1eb778
1 parent 3cfa906
commit a1eb778
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/open/text/embeddings/server/modal.py b/open/text/embeddings/server/modal.py
@@ -17,7 +17,7 @@
 memory = int(os.getenv("MEMORY", 2048)) 
 timeout = int(os.getenv("TIMEOUT", 600))
 keep_warm = int(os.getenv("KEEP_WARM", 1))
-@stub.function(image=image, cpu=cpu, memory=memory, timeout=timeout, keep_warm=keep_warm)
+@app.function(image=image, cpu=cpu, memory=memory, timeout=timeout, keep_warm=keep_warm)
 @asgi_app()
 def fastapi_app():
     from open.text.embeddings.server.app import create_app
@@ -27,4 +27,4 @@ def fastapi_app():
 
 
 if __name__ == "__main__":
-    stub.deploy("webapp")
+    app.deploy("webapp")