AhmedSSabir commited on
Commit
c6c0775
1 Parent(s): f3436c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -6,6 +6,10 @@ import re
6
  import os
7
  import gradio as gr
8
  import requests
 
 
 
 
9
 
10
  #url = "https://github.com/simonepri/lm-scorer/tree/master/lm_scorer/models"
11
  #resp = requests.get(url)
6
  import os
7
  import gradio as gr
8
  import requests
9
+ import torch
10
+ from transformers import GPT2Tokenizer, GPT2LMHeadModel
11
+ from torch.nn.functional import softmax
12
+ import numpy as np
13
 
14
  #url = "https://github.com/simonepri/lm-scorer/tree/master/lm_scorer/models"
15
  #resp = requests.get(url)