Upload app.py

app.py

@@ -1,7 +1,4 @@
 import spaces
-
-"""
-import spaces
 import bm25s
 import gradio as gr
 import json
@@ -19,12 +16,11 @@ import torch.nn.functional as F
 from datasets import concatenate_datasets, load_dataset, load_from_disk
 from huggingface_hub import hf_hub_download
 import random
-"""
-import gradio as gr
-import random
-import time
-from transformers import AutoModelForCausalLM
+from contextual import ContextualAI
+from openai import AzureOpenAI
 from transformers import pipeline, set_seed
+
+
 generator = pipeline('text-generation', model='gpt2', device_map="auto")
 
 @spaces.GPU