Update app.py

app.py

@@ -323,7 +323,7 @@ def phonetic_match(text, query, method='levenshtein_distance', apply_phonetic=Fa
 
 from typing import List, Union
 import torch
-from transformers import AutoTokenizer, AutoModelForSeq2Gen
+from transformers import AutoTokenizer
 import numpy as np
 from nltk.tokenize import word_tokenize
 from nltk.corpus import wordnet
@@ -375,7 +375,7 @@ def optimize_query(
                 model_max_length=128,  # Limit maximum sequence length
                 cache_dir="./model_cache"  # Cache models locally
             )
-            model = AutoModelForSeq2Gen.from_pretrained(
+            model = AutoModel.from_pretrained(
                 model_name,
                 low_cpu_mem_usage=True,  # Enable low memory usage
                 device_map="cpu"  # Explicitly set to CPU