Spaces:

srikanthp07
/

S21gpt

Runtime error

srikanthp07 commited on Oct 27, 2023

Commit

2130c07

1 Parent(s): 38668de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -187,7 +187,7 @@ class BigramLanguageModel(nn.Module):
 model = BigramLanguageModel()
 the_model = model.to(device)
-the_model.load_state_dict(torch.load('model.pth'))
 the_model.eval()
 def analyze_sentiment(text):

 model = BigramLanguageModel()
 the_model = model.to(device)
+the_model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
 the_model.eval()
 def analyze_sentiment(text):