sarim commited on
Commit
d58ef6e
·
1 Parent(s): 8cad885

update model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@
2
  import streamlit as st
3
  from pydantic_ai import Agent
4
  from pydantic_ai.models.groq import GroqModel
 
5
  import nest_asyncio
6
  import random
7
  import planet_model as customModel
@@ -14,7 +15,7 @@ api_key = os.getenv("api_key")
14
 
15
  random_number = random.randint(3000, 10000)
16
 
17
- model = GroqModel('llama-3.1-70b-versatile', api_key = api_key)
18
 
19
 
20
  async def onClick(vacationType,familyType,duration,purpose,interests):
 
2
  import streamlit as st
3
  from pydantic_ai import Agent
4
  from pydantic_ai.models.groq import GroqModel
5
+ from pydantic_ai.models.anthropic import AnthropicModel
6
  import nest_asyncio
7
  import random
8
  import planet_model as customModel
 
15
 
16
  random_number = random.randint(3000, 10000)
17
 
18
+ model = AnthropicModel('claude-3-5-sonnet-latest', api_key = api_key)
19
 
20
 
21
  async def onClick(vacationType,familyType,duration,purpose,interests):