better gpu

app.py

@@ -14,7 +14,7 @@ model.to(device)
 def output_generate(image):
     im = transform(image).unsqueeze(0).to(device)
     generated = model.generate(im, seq_len=20)
-    return open_clip.decode(generated[0]).split("<end_of_text>")[0].replace("<start_of_text>", "")
+    return open_clip.decode(generated[0].detach()).split("<end_of_text>")[0].replace("<start_of_text>", "")
 
 iface = gr.Interface(fn=output_generate, inputs=gr.Image(type="pil"), outputs="text")
 iface.launch()