Spaces:

yipengsun
/

AutoCitation

Sleeping

yipengsun commited on Jan 24

Commit

f1fefc1

verified ·

1 Parent(s): 3a37748

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -129,7 +129,7 @@ class CitationGenerator:
         self.xml_parser = ArxivXmlParser()
         self.async_context = AsyncContextManager()
         self.llm = ChatGoogleGenerativeAI(
-            model="gemini-1.5-flash",
             temperature=0.3,
             google_api_key=config.gemini_api_key,
             streaming=True

         self.xml_parser = ArxivXmlParser()
         self.async_context = AsyncContextManager()
         self.llm = ChatGoogleGenerativeAI(
+            model="gemini-2.0-flash-exp",
             temperature=0.3,
             google_api_key=config.gemini_api_key,
             streaming=True