darshan8950 commited on
Commit
7497699
·
verified ·
1 Parent(s): 6d8186b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -0
main.py CHANGED
@@ -9,6 +9,10 @@ llm = AutoModelForCausalLM.from_pretrained("TheBloke/Mistral-7B-v0.1-GGUF", mode
9
 
10
  app = Flask(__name__)
11
 
 
 
 
 
12
  @app.route('/recommend', methods=['POST'])
13
  def recommendation():
14
  content = request.json
 
9
 
10
  app = Flask(__name__)
11
 
12
+ @app.route('/')
13
+ def home():
14
+ return jsonify({"message": "Welcome to the Recommendation API!"})
15
+
16
  @app.route('/recommend', methods=['POST'])
17
  def recommendation():
18
  content = request.json