Update app.py

app.py

@@ -1,85 +1,6 @@
 import gradio as gr
-import time
-import logging
 
-# Set up logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-logger = logging.getLogger(__name__)
-
-def load_model(model_path):
-    logger.info(f"Attempting to load model from {model_path}...")
-    try:
-        # Try to load the model
-        logger.info("Before calling gr.load")
-        result = gr.load(model_path)
-        logger.info("After calling gr.load")
-        
-        # Add extensive debugging information
-        logger.info(f"Model loaded. Result type: {type(result)}")
-        
-        # If it's a tuple, inspect its contents
-        if isinstance(result, tuple):
-            logger.info(f"Result is a tuple of length {len(result)}")
-            for i, item in enumerate(result):
-                logger.info(f"Tuple item {i}: type={type(item)}")
-                
-                # If the item has attributes, log them
-                if hasattr(item, "__dict__"):
-                    logger.info(f"Item {i} attributes: {dir(item)}")
-                
-                # If it's a complex object, try to get more info
-                try:
-                    logger.info(f"Item {i} representation: {repr(item)}")
-                except:
-                    logger.info(f"Could not get representation of item {i}")
-        
-        logger.info(f"Successfully loaded demo from {model_path}.")
-        return result
-    except Exception as e:
-        logger.error(f"Failed to load demo from {model_path}: {e}")
-        # Get more detailed error information
-        import traceback
-        logger.error(f"Detailed error traceback: {traceback.format_exc()}")
-        return None
-
-# Load the demo
-demo = load_model("models/John6666/the-araminta-fv1-sdxl")
-#demo = load_model("models/John6666/the-araminta-experiment-fv2-sdxl")
-
-def generate_image(prompt, max_retries=5):
-    if demo is None:
-        return None, "Error: Demo failed to load."
-
-    for attempt in range(max_retries):
-        try:
-            logger.info(f"Attempt {attempt + 1}: Starting image generation with prompt: {prompt}")
-            start_time = time.time()
-            result = demo(prompt)
-            generation_time = time.time() - start_time
-            logger.info(f"Generation completed in {generation_time:.2f} seconds")
-            logger.info(result)
-            return result, f"Image generated successfully\nGeneration time: {generation_time:.2f} seconds"
-        except Exception as e:
-            logger.error(f"Error on attempt {attempt + 1}: {str(e)}")
-            if attempt < max_retries - 1:
-                logger.info(f"Retrying... ({attempt + 2}/{max_retries})")
-            else:
-                logger.error("Max retries reached. Giving up.")
-                return None, f"Failed to generate image after {max_retries} attempts. Last error: {str(e)}"
+with gr.Blocks(fill_height=True) as demo:
+    gr.load("models/John6666/the-araminta-fv1-sdxl", provider="hf-inference")
     
-    # This line should never be reached, but just in case:
-    return None, "Unexpected error occurred"
-
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=[
-        gr.Image(label="Generated Image"),
-        gr.Textbox(label="Output Information", lines=5)
-    ],
-    title="Text-to-Image Generation",
-    description="Enter a prompt to generate an image. The system will automatically retry up to 5 times if an error occurs."
-)
-
-logger.info("Launching the Gradio interface...")
-iface.launch()
+demo.launch()