Update myapp.py
Browse files
myapp.py
CHANGED
@@ -8,7 +8,7 @@ import os
|
|
8 |
from threading import RLock
|
9 |
from PIL import Image
|
10 |
|
11 |
-
|
12 |
|
13 |
lock = RLock()
|
14 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
@@ -55,7 +55,7 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
|
|
55 |
return None
|
56 |
|
57 |
# API function to perform inference
|
58 |
-
@
|
59 |
def generate_image():
|
60 |
data = request.get_json()
|
61 |
model_str = data['model_str']
|
@@ -74,4 +74,4 @@ def generate_image():
|
|
74 |
|
75 |
# Add this block to make sure your app runs when called
|
76 |
if __name__ == "__main__":
|
77 |
-
|
|
|
8 |
from threading import RLock
|
9 |
from PIL import Image
|
10 |
|
11 |
+
myapp = Flask(__name__)
|
12 |
|
13 |
lock = RLock()
|
14 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
|
55 |
return None
|
56 |
|
57 |
# API function to perform inference
|
58 |
+
@myapp.route('/generate-image', methods=['POST'])
|
59 |
def generate_image():
|
60 |
data = request.get_json()
|
61 |
model_str = data['model_str']
|
|
|
74 |
|
75 |
# Add this block to make sure your app runs when called
|
76 |
if __name__ == "__main__":
|
77 |
+
myapp.run(host='0.0.0.0', port=7860) # Run directly
|