Spaces:

pabloruizponce
/

in2IN

Runtime error

pabloruizponce commited on Oct 2

Commit

0b1bfdd

•

1 Parent(s): ec4c52c

removed cuda full

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,8 +29,7 @@ def generate(textI, texti1, texti2):
     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"),

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