Update app.py

app.py

@@ -67,17 +67,19 @@ def run_inference(model_name, prompt_text):
     ]
     subprocess.run(cmd)
 
-    # Find the latest generated file in the save directory
-    save_dir = "./output"  # Or the appropriate directory where files are saved
-    list_of_files = glob.glob(f'{save_dir}/*')  # You might need to adjust the pattern
-    latest_file = max(list_of_files, key=os.path.getctime)
+    save_dir = "./outputs/samples/"  # Örneğin, inference.py tarafından kullanılan kayıt dizini
+    list_of_files = glob.glob(f'{save_dir}/*')
+    if list_of_files:
+        latest_file = max(list_of_files, key=os.path.getctime)
+        return latest_file
+    else:
+        print("No files found in the output directory.")
+        return None
 
     # Clean up the temporary files
     os.remove(temp_file.name)
     os.remove(prompt_file.name)
 
-    return latest_file
-
 def main():
     gr.Interface(
         fn=run_inference,