Update app.py

app.py

@@ -3,13 +3,23 @@ import pip
 import os
 pip.main(['install', 'transformers'])
 pip.main(['install', 'torch'])
+pip.main(['install', 'pymongo'])
+import pymongo
 from transformers import pipeline
+from datetime import datetime
 model_name_or_path='dgnk007/eagle'
+generate=pipeline('text-generation',model=model_name_or_path)
+
+myclient = pymongo.MongoClient(os.environ['DB_URI'])
+mydb = myclient["eagle"]
+def store_at_db(prompt,response):
+    rawusage = mydb["rawusage"]
+    rawusage.insert_one({"prompt":prompt,"response":response,"current_time":datetime.now()})
 
 def generate_response(message):
     prompt_template=f"Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n {message}\n\n### Response:\n"
-    generate=pipeline('text-generation',model=model_name_or_path)
     response=generate(prompt_template,max_length=1024,return_full_text=False,eos_token_id=21017,pad_token_id=50256)
+    store_at_db(message,response[0]['generated_text'])
     return response[0]['generated_text']
 examples = [
     ["Give three tips for staying healthy."],