test

modules/app.py

@@ -86,7 +86,7 @@ async def ping():
 
         model_is_loading = True
 
-        import runtime
+        from . import runtime
 
         runtime.load_model_ckpt(ckpt_to_use=get_initial_model_to_load())
 
@@ -141,7 +141,7 @@ def save_model_to_config(model_name):
 
 @app.post('/image')
 def image(req : ImageRequest):
-    import runtime
+    from . import runtime
 
     r = Request()
     r.session_id = req.session_id
@@ -199,7 +199,7 @@ def stop():
         if model_is_loading:
             return {'ERROR'}
 
-        import runtime
+        from . import runtime
         runtime.stop_processing = True
 
         return {'OK'}
@@ -209,7 +209,7 @@ def stop():
 
 @app.get('/image/tmp/{session_id}/{img_id}')
 def get_image(session_id, img_id):
-    import runtime
+    from . import runtime
     buf = runtime.temp_images[session_id + '/' + img_id]
     buf.seek(0)
     return StreamingResponse(buf, media_type='image/jpeg')