Update myapp.py
Browse files
myapp.py
CHANGED
@@ -5,8 +5,8 @@ from huggingface_hub import InferenceClient
|
|
5 |
from io import BytesIO # For converting image to bytes
|
6 |
|
7 |
# Initialize the Flask app
|
8 |
-
|
9 |
-
CORS(
|
10 |
|
11 |
# Initialize the InferenceClient with your Hugging Face token
|
12 |
HF_TOKEN = os.environ.get("HF_TOKEN") # Ensure to set your Hugging Face token in the environment
|
@@ -22,7 +22,7 @@ def generate_image(prompt, seed=1, model="prompthero/openjourney-v4"):
|
|
22 |
return None
|
23 |
|
24 |
# Flask route for the API endpoint
|
25 |
-
@
|
26 |
def generate_api():
|
27 |
data = request.get_json()
|
28 |
|
@@ -52,4 +52,4 @@ def generate_api():
|
|
52 |
|
53 |
# Add this block to make sure your app runs when called
|
54 |
if __name__ == "__main__":
|
55 |
-
|
|
|
5 |
from io import BytesIO # For converting image to bytes
|
6 |
|
7 |
# Initialize the Flask app
|
8 |
+
myapp = Flask(__name__)
|
9 |
+
CORS(myapp) # Enable CORS for all routes
|
10 |
|
11 |
# Initialize the InferenceClient with your Hugging Face token
|
12 |
HF_TOKEN = os.environ.get("HF_TOKEN") # Ensure to set your Hugging Face token in the environment
|
|
|
22 |
return None
|
23 |
|
24 |
# Flask route for the API endpoint
|
25 |
+
@myapp.route('/generate_image', methods=['POST'])
|
26 |
def generate_api():
|
27 |
data = request.get_json()
|
28 |
|
|
|
52 |
|
53 |
# Add this block to make sure your app runs when called
|
54 |
if __name__ == "__main__":
|
55 |
+
myapp.run(host='0.0.0.0', port=7860) # Run directly if needed for testing
|