Spaces:

eagletiger1
/

dlagpt

Paused

Lin Chen commited on Aug 19, 2024

Commit

f1beaad

1 Parent(s): 29e9fb6

GPT-2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
 def get_bot_response(prompt):
-    pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-405B", use_auth_token = st.secrets['dla_test'])
     output = pipe(prompt, max_length=100)

     st.session_state.messages = []
 def get_bot_response(prompt):
+    pipe = pipeline("text-generation", model="gpt2")
     output = pipe(prompt, max_length=100)