From 6ba93652c65618a1ed9c89098cf3624bbc16e0bb Mon Sep 17 00:00:00 2001 From: Jeffrey Morgan Date: Sun, 25 Jun 2023 14:18:48 -0400 Subject: [PATCH] fix model lookup directory --- proto.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/proto.py b/proto.py index 46103a26..d8812e7a 100644 --- a/proto.py +++ b/proto.py @@ -52,7 +52,7 @@ def generate(): return Response("Model is required", status=400) if not prompt: return Response("Prompt is required", status=400) - if not os.path.exists(f"../models/{model}.bin"): + if not os.path.exists(f"./models/{model}.bin"): return {"error": "The model does not exist."}, 400 if model not in llms: @@ -74,6 +74,6 @@ def generate(): stream_with_context(stream_response()), mimetype="text/event-stream" ) - if __name__ == "__main__": app.run(debug=True, threaded=True, port=5001) + app.run()