Update llm_inference.py

llm_inference.py

@@ -22,7 +22,6 @@ class LLMInferenceNode:
             api_key=self.sambanova_api_key,
             base_url="https://api.sambanova.ai/v1",
         )
-        self.flux_client = Client("KingNish/Realtime-FLUX", hf_token=self.huggingface_token) 
 
     def generate_prompt(self, dynamic_seed, prompt_type, custom_input):
         """
@@ -242,19 +241,4 @@ Your output is only the caption itself, no comments or extra formatting. The cap
 
         except Exception as e:
             print(f"An error occurred: {e}")
-            return f"Error occurred while processing the request: {str(e)}"
-
-    def generate_image(self, prompt, seed=42, width=1024, height=1024):
-        try:
-            result = self.flux_client.predict(
-                prompt=prompt,
-                seed=seed,
-                width=width,
-                height=height,
-                api_name="/generate_image"
-            )
-            # Extract the image path from the result tuple
-            image_path = result[0]
-            return image_path
-        except Exception as e:
-            raise Exception(f"Error generating image: {str(e)}")
+            return f"Error occurred while processing the request: {str(e)}"