Update app.py

app.py

@@ -1,27 +1,16 @@
-import openai
-import gradio as gr
+from transformers import DistilBertTokenizer, DistilBertModel
+import torch
+tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased-distilled-squad')
+model = DistilBertModel.from_pretrained('distilbert-base-cased-distilled-squad')
 
-#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
+question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet"
 
-openai.api_key = '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'
 
-def openai_chat(prompt):
-    completions = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=2048,
-        n=1,
-        temperature=0.5,
-    )
 
-    message = completions.choices[0].text
-    return message.strip()
 
-def chatbot(input, history=[]):
-    output = openai_chat(input)
-    history.append((input, output))
-    return history, history
 
-gr.Interface( title = 'An Alternative to chatGPT - Built by Hrishikesh', fn = chatbot,
-             inputs = ["text",'state'],
-             outputs = ["chatbot",'state']).launch(debug = True )
+inputs = tokenizer(question, text, return_tensors="pt")
+with torch.no_grad():
+    outputs = model(**inputs)
+
+print(outputs)