Spaces:

dhmeltzer
/

qg_generation

Sleeping

dhmeltzer commited on May 23, 2023

Commit

01c57f3

1 Parent(s): ec6571c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def main():
     checkpoints = ['dhmeltzer/bart-large_askscience-qg',
               'dhmeltzer/flan-t5-base_askscience-qg',
-              'philschmid/flan-t5-xxl-sharded-fp16']
     headers = {"Authorization": f"Bearer {st.secrets['HF_token']}"}
     openai.api_key = st.secrets['OpenAI_token']

     checkpoints = ['dhmeltzer/bart-large_askscience-qg',
               'dhmeltzer/flan-t5-base_askscience-qg',
+              'google/flan-t5-xxl']
     headers = {"Authorization": f"Bearer {st.secrets['HF_token']}"}
     openai.api_key = st.secrets['OpenAI_token']