Geek7 commited on
Commit
4b8da63
·
verified ·
1 Parent(s): f06461c

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +4 -4
myapp.py CHANGED
@@ -8,8 +8,8 @@ import asyncio
8
  from threading import RLock
9
 
10
  # Initialize Flask app and enable CORS
11
- app = Flask(__name__)
12
- CORS(app)
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
- @app.route('/generate', methods=['POST'])
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
- app.run(debug=True)
 
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)