Update app.py

app.py

@@ -7,8 +7,8 @@ import wikipediaapi
 
 # Initialize Wikipedia API with a custom user-agent
 wiki_wiki = wikipediaapi.Wikipedia(
-    language='en',  # Specify the language as a keyword argument
-    user_agent='HumanizedTextApp/1.0 (kinelyaydenseo19@gmail.com)'  # Proper user agent format
+    language='en',
+    user_agent='HumanizedTextApp/1.0 (kinelyaydenseo19@gmail.com)'
 )
 
 # Function to fetch content from Wikipedia
@@ -27,7 +27,7 @@ embedder = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 
 # Fetch and create the corpus
 titles = [
-   "Crypto",
+    "Crypto",
     "Finance",
     "Technology",
     "Healthcare",
@@ -71,8 +71,8 @@ if st.button("Generate Humanized Text"):
         # Concatenate user input and context for model input
         input_text = f"{user_input} {context}"
         
-        # Tokenize input
-        inputs = tokenizer(input_text, return_tensors="pt", truncation=True, max_length=1024)
+        # Tokenize input and handle truncation
+        inputs = tokenizer(input_text, return_tensors="pt", max_length=1024, truncation=True)
 
         # Generate output
         outputs = model.generate(inputs.input_ids, max_length=2000, num_return_sequences=1)