Update main.py

main.py

@@ -6,11 +6,15 @@ from flask import Flask, request, jsonify, redirect, url_for
 from flask_cors import CORS
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
+from sqlalchemy import create_engine, Column, Integer, String, Boolean, ForeignKey
+from sqlalchemy.ext.declarative import declarative_base
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 connection_string = "postgresql://data_owner:PFAnX9oJp4wV@ep-green-heart-a78sxj65.ap-southeast-2.aws.neon.tech/figurecircle?sslmode=require"
 
+Base = declarative_base()
+
 class Mentor(Base):
     __tablename__ = 'mentors'