Update main.py

main.py

@@ -9,6 +9,7 @@ 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
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")