Update app.py

app.py

@@ -33,7 +33,7 @@ def inference(sample_name):
     }
 
 # List of sample file names
-sample_files = sorted(os.listdir(os.path.join(os.getcwd(),'samples','*.pth')))
+sample_files = sorted(os.listdir(os.path.join(os.getcwd(),'samples')))
 sample_names = [os.path.basename(x).split('.pth')[0] for x in sample_files]
 # Create Gradio interface
 dropdown = gr.inputs.Dropdown(choices=sample_names, label="Select a sample")