Commit 6ba93652 authored by Jeffrey Morgan's avatar Jeffrey Morgan
Browse files

fix model lookup directory

parent 0ce372f7
...@@ -52,7 +52,7 @@ def generate(): ...@@ -52,7 +52,7 @@ def generate():
return Response("Model is required", status=400) return Response("Model is required", status=400)
if not prompt: if not prompt:
return Response("Prompt is required", status=400) 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 return {"error": "The model does not exist."}, 400
if model not in llms: if model not in llms:
...@@ -74,6 +74,6 @@ def generate(): ...@@ -74,6 +74,6 @@ def generate():
stream_with_context(stream_response()), mimetype="text/event-stream" stream_with_context(stream_response()), mimetype="text/event-stream"
) )
if __name__ == "__main__": if __name__ == "__main__":
app.run(debug=True, threaded=True, port=5001) app.run(debug=True, threaded=True, port=5001)
app.run()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment