Update app.py

app.py

@@ -12,7 +12,8 @@ model, _, transform = open_clip.create_model_and_transforms(
 )
 model.to(device)
 model.eval()
-traced_model = torch.jit.trace(model, torch.zeros((1, 3, 64, 64)).to(device))
+traced_generate = torch.jit.trace(model.text_generator.generate, (torch.rand(1, 64).to(device),))
+
 
 def output_generate(image):
     im = transform(image).unsqueeze(0).to(device)