Explorar o código

fix model lookup directory

Jeffrey Morgan hai 1 ano
pai
achega
6ba93652c6
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      proto.py

+ 2 - 2
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()