Update app.py

app.py

@@ -37,6 +37,7 @@ def extend(input_text, num_return_sequences, max_size=20, top_k=50, top_p=0.95):
     if len(output_sequences.shape) > 2:
         output_sequences.squeeze_()
     generated_sequences = []
+    print(output_sequences)
     for generated_sequence_idx, generated_sequence in enumerate(output_sequences):        
         generated_sequence = generated_sequence.tolist()
         # Decode text
@@ -81,18 +82,13 @@ if __name__ == "__main__":
             print(f"device:{device}, n_gpu:{n_gpu}, random_seed:{random_seed}, maxlen:{max_len}, top_k:{top_k}, top_p:{top_p}")
             if len(text_area.strip()) == 0:
                 text_area = random.choice(suggested_text_list)
-            result = extend(input_text=text_area,                         
+            result = extend(input_text=text_area,
+                            num_return_sequences=int(num_return_sequences),                         
                             max_size=int(max_len),                   
                             top_k=int(top_k),
-                            top_p=float(top_p),
-                            num_return_sequences=int(num_return_sequences))
+                            top_p=float(top_p))
             print("Done length: " + str(len(result)) + " bytes") 
             #<div class="rtl" dir="rtl" style="text-align:right;">
-            st.markdown(f"<p dir=\"rtl\" style=\"text-align:right;\"> {result} </p>", unsafe_allow_html=True)
+            st.markdown(f"{result}", unsafe_allow_html=True)
             st.write("\n\nResult length: " + str(len(result)) + " bytes")
-            print(f"\"{result}\"")      
-    
-    st.markdown(    
-        """Hebrew text generation model (125M parameters) based on EleutherAI's gpt-neo architecture. Originally trained on a TPUv3-8 which was made avilable to me via the [TPU Research Cloud Program](https://sites.research.google/trc/)."""
-    )
-    st.markdown("<footer><hr><p style=\"font-size:14px\">Enjoy</p><p style=\"font-size:12px\">Created by <a href=\"https://linktr.ee/Norod78\">Doron Adler</a></p></footer> ", unsafe_allow_html=True)
+            print(f"\"{result}\"")