Upload new file

vanna_model2.py

+import pandas as pd
+import vanna as vn
+
+
+# Create login and get API key
+api_key = vn.get_api_key('saroj.dhiman@ociustechnologies.com') 
+vn.set_api_key(api_key)
+
+# Create models
+vn.create_model(model="data_warehose", db_type="Snowflake")
+vn.create_model(model="marketing_db", db_type="Postgres")
+vn.create_model(model="marketing_db", db_type="BigQuery")
+
+models = vn.get_models()
+
+  # Set model
+vn.set_model('chinook')
+
+# Connect to database
+url= "database.sqlite3"
+vn.connect_to_sqlite(url=url)
+
+# Check training data
+training_data = vn.get_training_data()
+training_data
+
+
+vn.ask(question='What are the most occured cancer in CancerData?')
+
+# training_data = vn.get_training_data()
+# training_data['question'].str.contains('total Stomach records in cancertable?').any()
+
+# vn.ask(question='give the most record of cancer are?')
\ No newline at end of file