Update app.py

app.py

@@ -27,7 +27,6 @@ def fine_tune_model():
 
 def upload_model(model_id, tokenizer):
     model_name = model_id[model_id.rfind('/')+1:]
-    print(model_name)
     model_repo_name = f"{profile}/{model_name}"
 
     login(token=os.environ["HF_TOKEN"])
@@ -40,12 +39,14 @@ def upload_model(model_id, tokenizer):
     )
 
     tokenizer.push_to_hub(model_repo_name)
-    
+
+    return model_repo_name
+
 def process(model_id, dataset):
     tokenizer = download_model(model_id)
-    upload_model(model_id, tokenizer)
+    model_repo_name = upload_model(model_id, tokenizer)
     
-    return "Processing completed"
+    return model_repo_name
 
 demo = gr.Interface(fn=process, 
                     inputs=[gr.Textbox(label = "Model ID", value = "meta-llama/Meta-Llama-3-8B-Instruct", lines = 1),