Update app.py

app.py

@@ -125,7 +125,7 @@ def train_model(dataset_name, model_name, epochs, batch_size, learning_rate, out
         hf_api.upload_folder(
             folder_path=output_dir,
             path_in_repo=".",
-            repo_id=model_name,
+            repo_id="sakaltcommunity/grape-small",
             token=hf_token
         )