Update app.py

app.py

@@ -33,7 +33,7 @@ def initialize_models():
         # Download model repository (only once)
         model_path = snapshot_download(
             repo_id="NightRaven109/CCSRModels",
-            token=os.environ['Read2']
+            token=os.environ.get('Read2', '')  # Use get() to avoid KeyError
         )
 
         # Set up default arguments
@@ -257,7 +257,7 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
     
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(label="Input Image")
+            input_image = gr.Image(label="Input Image", type="numpy")
             
             with gr.Accordion("Advanced Options", open=False):
                 prompt = gr.Textbox(label="Prompt", value=DEFAULT_VALUES["prompt"])
@@ -278,7 +278,7 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
                 submit_btn = gr.Button("Submit", variant="primary")
 
         with gr.Column():
-            output_image = gr.Image(label="Generated Image", type="pil", format="png")
+            output_image = gr.Image(label="Generated Image", type="pil")
 
     # Add examples
     gr.Examples(
@@ -329,4 +329,4 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()