rootp1 commited on
Commit
272eb13
·
verified ·
1 Parent(s): e80abd9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -13
app.py CHANGED
@@ -1,17 +1,11 @@
1
  from flask import Flask, request, jsonify
2
  from flask_cors import CORS
3
  from model import load_model, predict_species, get_label_names
4
- import os
5
- from dotenv import load_dotenv
6
-
7
- # Load environment variables
8
- load_dotenv()
9
 
10
  app = Flask(__name__)
11
 
12
  # Configure CORS with environment variables
13
- cors_origins = os.getenv('CORS_ORIGINS', 'http://localhost:3000').split(',')
14
- CORS(app, origins=cors_origins)
15
 
16
  model = load_model()
17
  label_names = get_label_names()
@@ -28,9 +22,5 @@ def predict():
28
  return jsonify({'error': str(e)}), 500
29
 
30
  if __name__ == '__main__':
31
- # Get configuration from environment variables
32
- host = os.getenv('FLASK_HOST', '127.0.0.1')
33
- port = int(os.getenv('FLASK_PORT', 5000))
34
- debug = os.getenv('FLASK_DEBUG', 'True').lower() == 'true'
35
-
36
- app.run(host=host, port=port, debug=debug)
 
1
  from flask import Flask, request, jsonify
2
  from flask_cors import CORS
3
  from model import load_model, predict_species, get_label_names
 
 
 
 
 
4
 
5
  app = Flask(__name__)
6
 
7
  # Configure CORS with environment variables
8
+ CORS(app, origins='https://green-gain.vercel.app/')
 
9
 
10
  model = load_model()
11
  label_names = get_label_names()
 
22
  return jsonify({'error': str(e)}), 500
23
 
24
  if __name__ == '__main__':
25
+ # Run only on localhost (loopback) at port 5000
26
+ app.run(host='127.0.0.1', port=5000, debug=True)