Update app.py

app.py

@@ -32,7 +32,7 @@ model = AutoModelForCausalLM.from_pretrained(
 '''
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-peft_model = AutoPeftModelForCausalLM.from_pretrained("ehristoforu/fd-lora-64x128", torch_dtype=torch.float16, trust_remote_code=True)
+peft_model = AutoPeftModelForCausalLM.from_pretrained("ehristoforu/fd-lora-16x32", torch_dtype=torch.float16, trust_remote_code=True)
 merged_model = peft_model.merge_and_unload()
 merged_model.save_pretrained("./coolqwen")
 #model.save_pretrained("./coolqwen")
@@ -46,7 +46,7 @@ api = HfApi()
 
 api.upload_folder(
     folder_path="./coolqwen",
-    repo_id="ehristoforu/fd-lora-merged-64x128",
+    repo_id="ehristoforu/fd-lora-merged-16x32",
     repo_type="model",
     token=HF_TOKEN,
 )