KingOfThoughtFleuren commited on
Commit
27660b8
·
verified ·
1 Parent(s): 98ff8a6

Update services/ontology_architect.py

Browse files
Files changed (1) hide show
  1. services/ontology_architect.py +1 -2
services/ontology_architect.py CHANGED
@@ -3,8 +3,7 @@
3
  import os
4
  import json
5
  import re
6
- import vertexai
7
- from vertexai.generative_models import GenerativeModel
8
 
9
  class OntologyArchitect:
10
  def __init__(self, models, data_directory):
 
3
  import os
4
  import json
5
  import re
6
+ import google.generativeai as genai
 
7
 
8
  class OntologyArchitect:
9
  def __init__(self, models, data_directory):