Spaces:

Ilvir
/

ilva

Runtime error

Ilvir commited on Jun 9, 2023

Commit

8188c2f

1 Parent(s): 3ed99c2

Update pages/gpt.py

Files changed (1) hide show

pages/gpt.py CHANGED Viewed

@@ -33,7 +33,7 @@ top_p = st.sidebar.slider('**Minimum total probability of top words:**', 0.4, 1.
 prompt = st.text_input('**Enter text 👇:**')
 if st.button('**Generate text**'):
     image_container = st.empty()
-    image_container.image("pict/wait.png", caption="that's so long!!!", use_column_width=True)
     with torch.inference_mode():
         prompt = tokenizer.encode(prompt, return_tensors='pt')
         out = model.generate(

 prompt = st.text_input('**Enter text 👇:**')
 if st.button('**Generate text**'):
     image_container = st.empty()
+    image_container.image("pict/wait.jpeg", caption="that's so long!!!", use_column_width=True)
     with torch.inference_mode():
         prompt = tokenizer.encode(prompt, return_tensors='pt')
         out = model.generate(