Geek7 commited on
Commit
7b2b25d
·
verified ·
1 Parent(s): 2c8a77f

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +3 -3
myapp.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  from threading import RLock
9
  from PIL import Image
10
 
11
- app = Flask(__name__)
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
- @app.route('/generate-image', methods=['POST'])
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
- app.run(host='0.0.0.0', port=7860) # Run directly
 
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