CyberCoder225 commited on
Commit
a58d40c
·
verified ·
1 Parent(s): 30256c9

Create brain.py

Browse files
Files changed (1) hide show
  1. brain.py +25 -0
brain.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from brain import MairaBrain
3
+
4
+ app = Flask(__name__)
5
+
6
+ # THE UPGRADE CONSTANTS
7
+ REPO_ID = "bartowski/Llama-3.2-1B-Instruct-GGUF"
8
+ FILENAME = "Llama-3.2-1B-Instruct-Q4_K_M.gguf"
9
+
10
+ maira = MairaBrain(REPO_ID, FILENAME)
11
+
12
+ @app.route("/", methods=["GET"])
13
+ def home(): return "Maira v6.0 is Online."
14
+
15
+ @app.route("/chat", methods=["POST"])
16
+ def chat():
17
+ data = request.json
18
+ user_id = data.get("user_id", "default_user")
19
+ user_input = data.get("message", "")
20
+
21
+ response = maira.get_response(user_id, user_input)
22
+ return jsonify({"response": response})
23
+
24
+ if __name__ == "__main__":
25
+ app.run(host="0.0.0.0", port=7860)