LeoWalker commited on
Commit
f76e0ff
·
verified ·
1 Parent(s): 84c8e82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -52,17 +52,16 @@ with col2:
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")
 
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
+ "Claude 3.5 Haiku": ChatAnthropic(model="claude-3-5-haiku-20241022"),
58
+ "Claude 3.5 Sonnet": ChatAnthropic(model="claude-3-5-sonnet-20241022"),
59
+ "Llama 3 8b": ChatGroq(model_name="llama3-8b-8192"),
60
+ "Llama 3 70b": ChatGroq(model_name="llama3-70b-8192"),
61
+ "Gemma 7b": ChatGroq(model_name="gemma-7b-it"),
62
  "Mixtral 8x7b": ChatGroq(model_name="mixtral-8x7b-32768"),
63
  "Gemini 1.5 Pro": ChatGoogleGenerativeAI(model="gemini-1.5-pro"),
64
+ "Gemini 1.5 Flash": ChatGoogleGenerativeAI(model="gemini-1.5-flash"),
 
65
  }
66
 
67
  uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")