wt002 commited on
Commit
0b35738
·
verified ·
1 Parent(s): 8b86d71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import requests
8
  from typing import List, Dict, Union
9
  import pandas as pd
10
  import wikipediaapi
11
- from google import genai
12
 
13
  load_dotenv()
14
 
@@ -39,7 +39,7 @@ class BasicAgent:
39
  return fixed_answer
40
 
41
  def answer_question(self, prompt: str) -> str:
42
- from google import genai
43
 
44
  client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
45
 
 
8
  from typing import List, Dict, Union
9
  import pandas as pd
10
  import wikipediaapi
11
+ from google.genarativeai import genai
12
 
13
  load_dotenv()
14
 
 
39
  return fixed_answer
40
 
41
  def answer_question(self, prompt: str) -> str:
42
+ from google.genarativeai import genai
43
 
44
  client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
45