Hugging Face's logo

Spaces:
APRG
/
Final_Assignment_Template
Running

App Files Community
APRG commited on
Commit
f963b6a
·
verified ·
1 Parent(s): 763a433

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -46,8 +46,10 @@ class BasicAgent:
46
 
47
  REQUEST:
48
  """
49
- llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
50
- summary = llm.invoke(prompt+question)
 
 
51
  return summary
52
 
53
  def __call__(self, question: str) -> str:
 
46
 
47
  REQUEST:
48
  """
49
+ llm = LiteLLMModel(
50
+ model_id="gemini-2.0-flash", api_key=os.getenv("GEMINI_API_KEY")
51
+ )
52
+ summary = llm.run(prompt+question)
53
  return summary
54
 
55
  def __call__(self, question: str) -> str: