Spaces:

pabloruizponce
/

in2IN

Runtime error

pabloruizponce commited on Oct 2

Commit

ec4c52c

•

1 Parent(s): e00c0aa

Removed cuda (for now)

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,8 @@ def generate(textI, texti1, texti2):
     return filename
 model = AutoModel.from_pretrained("pabloruizponce/in2IN", trust_remote_code=True)
-model.to("cuda")
 demo = gr.Interface(fn=generate,
                     inputs=[gr.Text(label="Interaction Description"),
                             gr.Text(label="Individual1 Description"),

     return filename
 model = AutoModel.from_pretrained("pabloruizponce/in2IN", trust_remote_code=True)
+if torch.cuda.is_available():
+    model.to("cuda")
 demo = gr.Interface(fn=generate,
                     inputs=[gr.Text(label="Interaction Description"),
                             gr.Text(label="Individual1 Description"),