sachitksh123 commited on
Commit
8f76f02
·
verified ·
1 Parent(s): 353edc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ session.verify = False # Disable SSL verification
15
  HUGGINGFACE_TOKEN = os.environ.get("hf_token") # Set the default to empty if not found
16
 
17
  # Set up the text generation pipeline with the token
18
- pipe = pipeline("text-generation", model="mattshumer/Reflection-Llama-3.1-70B",
19
  use_auth_token=HUGGINGFACE_TOKEN, request_session=session)
20
 
21
  # Streamlit application
 
15
  HUGGINGFACE_TOKEN = os.environ.get("hf_token") # Set the default to empty if not found
16
 
17
  # Set up the text generation pipeline with the token
18
+ pipe = pipeline("text-generation", model="mistralai/Mistral-7B-v0.1",
19
  use_auth_token=HUGGINGFACE_TOKEN, request_session=session)
20
 
21
  # Streamlit application