Update app.py

app.py

@@ -26,13 +26,10 @@ text_in = st.text_input('Введите текст:')
 start = st.button('Start:')
 if start:
     for sample in text_in:
-         data = tokenizer(sample, return_tensors="pt")
-         data = {k: v.to(model.device) for k, v in data.items()}
-         output_ids = model.generate(
-              **data,
-              generation_config=generation_config
-    )[0]
-    print(tokenizer.decode(data["input_ids"][0].tolist()))
-    print(tokenizer.decode(output_ids.tolist()))
-    print("====================")
+        data = tokenizer(sample, return_tensors="pt")
+        data = {k: v.to(model.device) for k, v in data.items()}
+        output_ids = model.generate(**data,generation_config=generation_config)[0]
+    #print(tokenizer.decode(data["input_ids"][0].tolist()))
+    #print(tokenizer.decode(output_ids.tolist()))
+    #print("====================")
     st.write("Результат:", tokenizer.decode(data["input_ids"][0].tolist()),tokenizer.decode(output_ids.tolist()))