Update app.py

app.py

@@ -67,10 +67,9 @@ def load_model(file_name: str, path:str,device: torch.device) -> nn.Module:
         model(z, label, force_fp32=True)
     return model
    
+#def image_create():
 def greet(name):
     return "Hello " + name + "!"
-
-
 def main():
     #torch.cuda.init()
     #if torch.cuda.is_initialized():
@@ -82,10 +81,11 @@ def main():
     args = ProjectorArguments().parse(
         args=[str(input_path)],
         namespace=Namespace(
-            # spectral_sensitivity=spectral_sensitivity,
-            encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
-            # encoder_name=spectral_sensitivity,
-            # gaussian=gaussian_radius,
+            spectral_sensitivity=spectral_sensitivity,
+            encoder_ckpt=f"checkpoint/encoder/c
+              heckpoint_{spectral_sensitivity}.pt",
+            encoder_name=spectral_sensitivity,
+            gaussian=gaussian_radius,
             log_visual_freq=1000,
             input='text',
         ))
@@ -115,7 +115,6 @@ def main():
     #)
     demo = gr.Interface(fn=greet, inputs="text", outputs="text")
     demo.launch()
-    
 if __name__ == '__main__':
     main()