Fix some bugs

README.md

@@ -1,7 +1,7 @@
 ---
 title: GPT2 Persian
 emoji: 🤘
-colorFrom: orange
+colorFrom: yellow
 colorTo: red
 sdk: streamlit
 app_file: app.py

app.py

@@ -33,7 +33,7 @@ class TextGeneration:
         if not self.debug:
             generation_kwargs["num_return_sequences"] = 1
 
-            max_length = len(self.tokenizer(prompt)["input_ids"][0]) + generation_kwargs["max_length"]
+            max_length = len(self.tokenizer(prompt)["input_ids"]) + generation_kwargs["max_length"]
             generation_kwargs["max_length"] = max_length
 
             return self.generator(
@@ -53,8 +53,8 @@ def load_text_generator():
 
 def main():
     st.set_page_config(
-        page_title="X",
-        page_icon="🧊",
+        page_title="GPT2 - Persian",
+        page_icon="🤘",
         layout="wide",
         initial_sidebar_state="expanded"
     )