Update app.py

app.py

@@ -14,7 +14,7 @@ def process(model_id, dataset):
     model.save_pretrained(model_path)
     login(token=os.environ["HF_TOKEN"])
     api = HfApi()
-    model_repo_name = "bstraehle/gemma-2b"   
+    model_repo_name = "bstraehle/Meta-Llama-3-8B"
     
     #Create Repo in Hugging Face
     print("333")
@@ -34,7 +34,7 @@ def process(model_id, dataset):
     return "Done"
 
 demo = gr.Interface(fn=process, 
-                    inputs=[gr.Textbox(label = "Model ID", value = "google/gemma-2b", lines = 1),
+                    inputs=[gr.Textbox(label = "Model ID", value = "meta-llama/Meta-Llama-3-8B", lines = 1), # google/gemma-2b
                             gr.Textbox(label = "Dataset", value = "imdb", lines = 1)],
                     outputs=[gr.Textbox(label = "Completion")])
 demo.launch()