Update app.py

app.py

@@ -207,21 +207,21 @@ elif selected_model == "Генерация текста GPT-моделью по
     num_generations = st.slider('Количество генераций', 1, 10, 2, step=1)
     
     if st.button('Сделать гороскоп'):
-    with st.spinner('Генерация текста...'):
-        start_time = time.time()
-        input_ids = tokenizer.encode(user_text_input, return_tensors="pt").to(device)
-        model.eval()
-        with torch.no_grad():
-            out = model.generate(
-                input_ids,
-                do_sample=True,
-                num_beams=num_generations,
-                temperature=temperature,
-                top_p=0.9,
-                max_length=max_gen_length,
-            )
-        generated_text = tokenizer.decode(out[0], skip_special_tokens=True)
-        end_time = time.time()
-        prediction_time = end_time - start_time
-
-        st.write(f'{generated_text}')
+        with st.spinner('Генерация текста...'):
+            start_time = time.time()
+            input_ids = tokenizer.encode(user_text_input, return_tensors="pt").to(device)
+            model.eval()
+            with torch.no_grad():
+                out = model.generate(
+                    input_ids,
+                    do_sample=True,
+                    num_beams=num_generations,
+                    temperature=temperature,
+                    top_p=0.9,
+                    max_length=max_gen_length,
+                )
+            generated_text = tokenizer.decode(out[0], skip_special_tokens=True)
+            end_time = time.time()
+            prediction_time = end_time - start_time
+    
+            st.write(f'{generated_text}')