Spaces:

ssocean
/

Newborn_Article_Impact_Predict

Running on Zero

ssocean commited on Dec 5, 2024

Commit

1bff2f0

verified ·

1 Parent(s): b404115

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model.eval()
 @spaces.GPU
 def predict(title, abstract):
     text = f'''Given a certain paper, Title: {title}\n Abstract: {abstract}. \n Predict its normalized academic impact (between 0 and 1):'''
     inputs = tokenizer(text, return_tensors="pt")
     with torch.no_grad():

 @spaces.GPU
 def predict(title, abstract):
+    model.to(device)
     text = f'''Given a certain paper, Title: {title}\n Abstract: {abstract}. \n Predict its normalized academic impact (between 0 and 1):'''
     inputs = tokenizer(text, return_tensors="pt")
     with torch.no_grad():