Spaces:

yeq6x
/

Image2Body_gradio

Running on Zero

yeq6x commited on Jul 7

Commit

2e48cc7

•

1 Parent(s): 5653d38

templates, port

Files changed (4) hide show

Dockerfile CHANGED Viewed

@@ -46,7 +46,5 @@ RUN echo '#!/bin/bash\nredis-server --daemonize yes\npython app.py' > /app/start
 RUN useradd -m appuser && chown -R appuser:appuser /app
 USER appuser
-EXPOSE 5000
 # 起動コマンドを変更
 CMD ["/app/start.sh"]

 RUN useradd -m appuser && chown -R appuser:appuser /app
 USER appuser
 # 起動コマンドを変更
 CMD ["/app/start.sh"]

app.py CHANGED Viewed

@@ -24,7 +24,6 @@ from gevent import pywsgi
 from geventwebsocket.handler import WebSocketHandler
 app = Flask(__name__)
-# app.secret_key = 'super_secret_key'
 CORS(app)
 socketio = SocketIO(app, cors_allowed_origins="*")
@@ -207,7 +206,7 @@ def handle_disconnect():
 # ルートパスのGETリクエストに対するハンドラ
 @app.route('/', methods=['GET'])
 def root():
-    return jsonify({"status": "ok", "message": "Server is running"}), 200
 # process_refined のエンドポイント
 @app.route('/process_refined', methods=['POST'])
@@ -260,12 +259,11 @@ if __name__ == '__main__':
     parser.add_argument('--use_local', action='store_true', help='Use local model')
     parser.add_argument('--use_gpu', action='store_true', help='Set to True to use GPU but if not available, it will use CPU')
     parser.add_argument('--use_dotenv', action='store_true', help='Use .env file for environment variables')
-    parser.add_argument('--debug', action='store_true', help='Run in debug mode')
     args = parser.parse_args()
-    # initialize(args.use_local, args.use_gpu, args.use_dotenv)
-    port = int(os.environ['PORT']) if 'PORT' in os.environ else 5000
     print(f"Starting server on port {port}")
     server = pywsgi.WSGIServer(('0.0.0.0', port), app, handler_class=WebSocketHandler)
     server.serve_forever()

 from geventwebsocket.handler import WebSocketHandler
 app = Flask(__name__)
 CORS(app)
 socketio = SocketIO(app, cors_allowed_origins="*")
 # ルートパスのGETリクエストに対するハンドラ
 @app.route('/', methods=['GET'])
 def root():
+    return render_template("index.html")
 # process_refined のエンドポイント
 @app.route('/process_refined', methods=['POST'])
     parser.add_argument('--use_local', action='store_true', help='Use local model')
     parser.add_argument('--use_gpu', action='store_true', help='Set to True to use GPU but if not available, it will use CPU')
     parser.add_argument('--use_dotenv', action='store_true', help='Use .env file for environment variables')
     args = parser.parse_args()
+    initialize(args.use_local, args.use_gpu, args.use_dotenv)
+    port = int(os.environ.get('PORT', 7860))
     print(f"Starting server on port {port}")
     server = pywsgi.WSGIServer(('0.0.0.0', port), app, handler_class=WebSocketHandler)
     server.serve_forever()

requirements.txt CHANGED Viewed

@@ -15,7 +15,6 @@ Pillow==10.3.0
 Requests==2.32.3
 tensorflow==2.16.1
 transforms==0.2.1
-transformers==4.42.3
 tokenizers
 pytorch_lightning
 python-dotenv

 Requests==2.32.3
 tensorflow==2.16.1
 transforms==0.2.1
 tokenizers
 pytorch_lightning
 python-dotenv

templates/index.html ADDED Viewed

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>{{ title }}</title>
+</head>
+<body>
+    <h1>Welcome to our website</h1>
+    <p>This is the home page rendered using a Flask template.</p>
+</body>
+</html>