JRQi commited on
Commit
71c9215
1 Parent(s): 05e2529

Update game3.py

Browse files
Files changed (1) hide show
  1. game3.py +1 -0
game3.py CHANGED
@@ -6,6 +6,7 @@ import gradio as gr
6
  import numpy as np
7
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
  from transformers import pipeline
 
9
 
10
  def read3(num_selected_former):
11
  fname = 'data3_convai2_inferred.txt'
 
6
  import numpy as np
7
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
  from transformers import pipeline
9
+ import torch
10
 
11
  def read3(num_selected_former):
12
  fname = 'data3_convai2_inferred.txt'