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 = f"bstraehle/{model_id}"   
+    model_repo_name = "bstraehle/gemma-2b"   
     
     #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-7b", lines = 1),
+                    inputs=[gr.Textbox(label = "Model ID", value = "google/gemma-2b", lines = 1),
                             gr.Textbox(label = "Dataset", value = "imdb", lines = 1)],
                     outputs=[gr.Textbox(label = "Completion")])
 demo.launch()