Update app.py

app.py

@@ -23,6 +23,7 @@ from scripts.text_web_browser import (
 )
 from scripts.visual_qa import visualizer
 from scripts.text_cleaner_tool import TextCleanerTool
+from scripts.flux_lora_tool import FluxLoRATool
 
 from smolagents import (
     CodeAgent,
@@ -161,7 +162,6 @@ class ModelManager:
                     device_map="auto",
                     max_new_tokens=1000,
                 )
-                
 
             else:
                 raise ValueError(f"Invalid inference type: {chosen_inference}")
@@ -210,7 +210,7 @@ class ToolRegistry:
             )
         except Exception as e:
             print(f"✗ Couldn't initialize image generation tool: {e}")
-            raise
+        return FluxLoRATool
 
 
 # ------------------------ Agent Creation and Execution ------------------------
@@ -650,4 +650,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-