revert

app.py

@@ -14,11 +14,9 @@ from upsampling import get_pipeline, upscale_image
 
 import gradio as gr
 
-from huggingface_hub import login
+# from huggingface_hub import login
+
 
-login(token='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')
-# HF_TOKEN = os.getenv('HF_TOKEN')
-# hf_api = HfApi(token=HF_TOKEN)
 
 # fix randomness
 th.manual_seed(0)
@@ -167,9 +165,10 @@ def decompose_image_demo(im, model):
 def combine_images_demo(im1, im2, model):
     sample_method = 'ddim'
     result = combine_components_slice(MODELS[model], GD[sample_method], im1, im2, indices='1,0,1,0', sample_method=sample_method, num_images=1, device=device)
-    result = Image.fromarray(result.permute(1, 2, 0).numpy())
-    if model == 'CelebA-HQ':
-        return upscale_image(result, pipe)
+    result = result.permute(1, 2, 0).numpy()
+    # result = Image.fromarray(result.permute(1, 2, 0).numpy())
+    # if model == 'CelebA-HQ':
+    #     return upscale_image(result, pipe)
     return result
 
 
@@ -200,7 +199,7 @@ MODELS = {
     'CelebA-HQ': celeb_model
 }
 
-pipe = get_pipeline()
+# pipe = get_pipeline()
 
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -265,7 +264,7 @@ with gr.Blocks() as demo:
             
 
         with gr.Column(scale=1):
-            comb_output = gr.Image(type='pil')
+            comb_output = gr.Image(type='numpy')
             comb_button = gr.Button("Generate")