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