Update main.py

main.py

@@ -4,12 +4,14 @@ import random
 
 from flask import Flask, request, jsonify, redirect, url_for
 from flask_cors import CORS
-from sqlalchemy import create_engine, Column, Integer, String, Boolean, ForeignKey, LargeBinary
+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
+from sqlalchemy import LargeBinary
 from sqlalchemy.orm import sessionmaker, relationship
 
-# Use a free AI model instead of Mixtral-8x7B
-client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.1")  # Free model
+client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 connection_string = "postgresql://neondb_owner:Pl8cWUu0iLHn@ep-tiny-haze-a1w7wrrg.ap-southeast-1.aws.neon.tech/neondb?sslmode=require"