Shankarm08 commited on
Commit
6ec5a5c
·
verified ·
1 Parent(s): be332a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import pipeline
7
  # Load the Hugging Face model (GPT-2 in this case)
8
  @st.cache_resource
9
  def load_model():
10
- return pipeline("text-generation", model="gpt2")
11
 
12
  # Function to return the response from Hugging Face model
13
  def load_answer(question):
 
7
  # Load the Hugging Face model (GPT-2 in this case)
8
  @st.cache_resource
9
  def load_model():
10
+ return pipeline("text-generation", model="gpt-4o")
11
 
12
  # Function to return the response from Hugging Face model
13
  def load_answer(question):