LM-Explanation-Demo

Runtime error

JRQi commited on Sep 27, 2023

Commit

71c9215

•

1 Parent(s): 05e2529

Update game3.py

Files changed (1) hide show

game3.py CHANGED Viewed

@@ -6,6 +6,7 @@ import gradio as gr
 import numpy as np
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
 def read3(num_selected_former):
     fname = 'data3_convai2_inferred.txt'

 import numpy as np
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
+import torch
 def read3(num_selected_former):
     fname = 'data3_convai2_inferred.txt'