Update app.py

app.py

@@ -131,7 +131,7 @@ transform_resize = A.Compose([
 
 # Load model
 generator_model = Generator()
-generator_model.load_state_dict(torch.load('age-transformation/large-aging-model.h5',map_location=torch.device(device)))
+generator_model.load_state_dict(torch.load('large-aging-model.h5',map_location=torch.device(device)))
 generator_model.to(device)
 #generator_model.eval()
 print("")
@@ -184,8 +184,8 @@ output_image = gr.Image(label="Output Image", type="pil")
 input_image.style(height=512, width=512)
 output_image.style(height=512, width=512)
 
-input_age = gr.Slider(label="Input Age")
-output_age = gr.Slider(label="Output Age")
+input_age = gr.Slider(label="Current Age")
+output_age = gr.Slider(label="Desired Age")
 
 
 # Define the function to be called when the button is pressed
@@ -203,4 +203,4 @@ def process_image(input_img,input_age,output_age):
 
 
 # Create the Gradio interface
-gr.Interface(fn=process_image, inputs=[input_image,input_age,output_age], outputs=output_image, title="Image Crop and Process").launch(debug=True)
+gr.Interface(fn=process_image, inputs=[input_image,input_age,output_age], outputs=output_image, title="Age Transformation").launch(debug=True)