Update app_flash.py

app_flash.py

@@ -116,7 +116,7 @@ def push_flashpack_model_to_hf(model, hf_repo: str):
 # ============================================================
 def train_flashpack_model(
     dataset_name: str = "rahul7star/prompt-enhancer-dataset",
-    max_encode: int = 5000,
+    max_encode: int = 1000,
     hidden_dim: int = 1024,
     push_to_hub: bool = True,
     hf_repo: str = "rahul7star/FlashPack"
@@ -152,7 +152,7 @@ def train_flashpack_model(
     # Loss & optimizer
     criterion = nn.CosineSimilarity(dim=1)
     optimizer = optim.Adam(model.parameters(), lr=1e-3)
-    max_epochs = 30
+    max_epochs = 50
     batch_size = 32
     n = short_embeddings.shape[0]