update examples

app.py

@@ -61,7 +61,11 @@ def accelerate_train_lora(steps, images):
       --max_train_steps={int(steps)}') #,shell=True) #30000
     print("*********** completing accelerate_train_lora ***********")
     #lora_trained_weights = "./output_example/lora_weight.pt" 
-    return f"{OUTPUT_DIR}/lora_weight.pt"
+    for file in os.listdir(f"{OUTPUT_DIR}"):
+        if file.endswith(".pt"):
+            print(os.path.join(f"{OUTPUT_DIR}", file))
+            return file 
+    #return f"{OUTPUT_DIR}/*.pt"
     
 with gr.Blocks() as demo:
     gr.Markdown("""<h1><center>LORA - Low-rank Adaptation for Fast Text-to-Image Diffusion Fine-tuning</center></h1>