Upload app.py with huggingface_hub

app.py

@@ -137,8 +137,8 @@ def augment(sources, targets, factor=3):
     return aug_s, aug_t
 
 
-@spaces.GPU(duration=300)
-def train_model(epochs=100, batch_size=128, augment_factor=3):
+@spaces.GPU(duration=600)
+def train_model(epochs=100, batch_size=256, augment_factor=2):
     """Train seq2seq on GPU."""
     # Load data
     with open("expression_pairs.json") as f: