Update myapp.py
Browse files
myapp.py
CHANGED
@@ -8,8 +8,8 @@ import asyncio
|
|
8 |
from threading import RLock
|
9 |
|
10 |
# Initialize Flask app and enable CORS
|
11 |
-
|
12 |
-
CORS(
|
13 |
|
14 |
lock = RLock()
|
15 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
@@ -50,7 +50,7 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
|
|
50 |
print(e)
|
51 |
return None
|
52 |
|
53 |
-
@
|
54 |
def generate():
|
55 |
data = request.json
|
56 |
model_str = data.get('model')
|
@@ -66,4 +66,4 @@ def generate():
|
|
66 |
return jsonify({"error": "Image generation failed"}), 500
|
67 |
|
68 |
if __name__ == '__main__':
|
69 |
-
|
|
|
8 |
from threading import RLock
|
9 |
|
10 |
# Initialize Flask app and enable CORS
|
11 |
+
myapp = Flask(__name__)
|
12 |
+
CORS(myapp)
|
13 |
|
14 |
lock = RLock()
|
15 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
|
50 |
print(e)
|
51 |
return None
|
52 |
|
53 |
+
@myapp.route('/generate', methods=['POST'])
|
54 |
def generate():
|
55 |
data = request.json
|
56 |
model_str = data.get('model')
|
|
|
66 |
return jsonify({"error": "Image generation failed"}), 500
|
67 |
|
68 |
if __name__ == '__main__':
|
69 |
+
myapp.run(debug=True)
|