Update app.py

app.py

@@ -15,14 +15,17 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 
 profile = "bstraehle"
 
-def download_model(model_id)
+def download_model(model_id):
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForCausalLM.from_pretrained(model_id)
     model.save_pretrained(model_id)
 
     return tokenizer
 
-def upload_model(model_id, tokenizer)
+def fine_tune:
+    return ""
+
+def upload_model(model_id, tokenizer):
     model_name = model_id[model_id.rfind('/')+1:]
     print(model_name)
     model_repo_name = f"{profile}/{model_name}"
@@ -42,7 +45,7 @@ def process(model_id, dataset):
     tokenizer = download_model(model_id)
     upload_model(model_id, tokenizer)
     
-    return "Done"
+    return "Processing completed"
 
 demo = gr.Interface(fn=process, 
                     inputs=[gr.Textbox(label = "Model ID", value = "meta-llama/Meta-Llama-3-8B-Instruct", lines = 1),