Geek7 commited on
Commit
0968c17
·
verified ·
1 Parent(s): bdf7f63

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +5 -5
myapp.py CHANGED
@@ -7,8 +7,8 @@ import asyncio
7
  from threading import RLock
8
 
9
  # Initialize Flask app and enable CORS
10
- myapp = Flask(__name__)
11
- CORS(myapp)
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
- @myapp.route('/generate', methods=['POST'])
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__ == "__main__":
73
- myapp.run(host='0.0.0.0', port=7860)
 
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)