Update app.py

app.py

@@ -16,7 +16,8 @@ text_inputs = torch.cat([clip.tokenize(f"a photo of a {c}") for c in cifar100.cl
 def generateOutput(source):
     # Prepare the inputs
     # image, class_id = cifar100[3637]
-    image_input = preprocess(source).unsqueeze(0).to(device)
+    image = Image.fromarray(image.astype('uint8'), 'RGB')
+    image_input = preprocess(image).unsqueeze(0).to(device)
 
     with torch.no_grad():
         image_features = model.encode_image(image_input)