LeoWalker commited on
Commit
e1d2667
·
verified ·
1 Parent(s): d708c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -52,15 +52,17 @@ with col2:
52
  """)
53
 
54
  llm_dict = {
55
- "GPT 3.5 turbo": ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo-0125"),
56
- "GPT 4o": ChatOpenAI(temperature=0, model_name="gpt-4o"),
57
- "Anthropic 3.5 Sonnet": ChatAnthropic(model="claude-3-5-sonnet-20240620"),
58
- "Llama 3 8b": ChatGroq(model_name="llama3-8b-8192"),
59
- "Llama 3 70b": ChatGroq(model_name="llama3-70b-8192"),
60
- "Gemma 7b": ChatGroq(model_name="gemma-7b-it"),
 
61
  "Mixtral 8x7b": ChatGroq(model_name="mixtral-8x7b-32768"),
62
  "Gemini 1.5 Pro": ChatGoogleGenerativeAI(model="gemini-1.5-pro"),
63
- "Gemini 1.5 Flash": ChatGoogleGenerativeAI(model="gemini-1.5-flash"),
 
64
  }
65
 
66
  uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
 
52
  """)
53
 
54
  llm_dict = {
55
+ "GPT-4o-mini": ChatOpenAI(temperature=0, model_name="gpt-4o-mini"),
56
+ "GPT-4o": ChatOpenAI(temperature=0, model_name="gpt-4o"),
57
+ "Anthropic 3.5 Haiku": ChatAnthropic(model="claude-3-5-haiku-latest"),
58
+ "Anthropic 3.5 Sonnet": ChatAnthropic(model="claude-3-5-sonnet-latest"),
59
+ "Llama 3.1 8b instant": ChatGroq(model_name="llama-3.1-8b-instant"),
60
+ "Llama 3.3 70b": ChatGroq(model_name="llama-3.3-70b-versatile"),
61
+ "Gemma 2 9b": ChatGroq(model_name="gemma2-9b-it"),
62
  "Mixtral 8x7b": ChatGroq(model_name="mixtral-8x7b-32768"),
63
  "Gemini 1.5 Pro": ChatGoogleGenerativeAI(model="gemini-1.5-pro"),
64
+ "Gemini 2.0 Flash": ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp"),
65
+ "Gemini 1.5 Flash 8B": ChatGoogleGenerativeAI(model="gemini-1.5-flash-8b")
66
  }
67
 
68
  uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")