Update app.py

app.py

@@ -3,8 +3,11 @@ import torch
 import random
 from unidecode import unidecode
 import re
+import os
+from tqdm import tqdm
+import requests
 from samplings import top_p_sampling, top_k_sampling, temperature_sampling
-from transformers import GPT2Model, GPT2LMHeadModel, PreTrainedModel
+from transformers import GPT2Config, GPT2Model, GPT2LMHeadModel, PreTrainedModel
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 PATCH_LENGTH = 128      # Patch Length
@@ -368,8 +371,6 @@ def generate_abc(prompt,
 
     print("\n"+" OUTPUT TUNES ".center(60, "#"))
 
-    start_time = time.time()
-
     for i in range(num_tunes):
         tune = "X:"+str(i+1) + "\n" + prompt
         lines = re.split(r'(\n)', tune)