Spaces:

vvv-knyazeva
/

NLP_project

Runtime error

vvv-knyazeva commited on Jun 8, 2023

Commit

76e122c

1 Parent(s): 119f843

Update gpt_v1.py

Files changed (1) hide show

gpt_v1.py CHANGED Viewed

@@ -17,7 +17,7 @@ model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
 prompt = st.text_input('Введите текст prompt:')
 length = st.slider('Длина генерируемой последовательности:', 1, 256, 16)
 num_samples = st.slider('Число генераций:', 1, 4, 1)
-temperature = st.slider('Температура:', 0.5, 3.0, 0.5)
 def generate_text(model, tokenizer, prompt, length, num_samples, temperature):

 prompt = st.text_input('Введите текст prompt:')
 length = st.slider('Длина генерируемой последовательности:', 1, 256, 16)
 num_samples = st.slider('Число генераций:', 1, 4, 1)
+temperature = st.slider('Температура:', 1.0, 5.0, 1.0)
 def generate_text(model, tokenizer, prompt, length, num_samples, temperature):