arjunanand13 commited on
Commit
1c1f5db
·
verified ·
1 Parent(s): c5772bd

Update qa_bot_chatgpt.py

Browse files
Files changed (1) hide show
  1. qa_bot_chatgpt.py +1 -0
qa_bot_chatgpt.py CHANGED
@@ -7,6 +7,7 @@ import torch
7
  from huggingface_hub import login
8
  import ast
9
  from openai import OpenAI
 
10
  class QAInfer:
11
  def __init__(self):
12
  torch.cuda.empty_cache()
 
7
  from huggingface_hub import login
8
  import ast
9
  from openai import OpenAI
10
+ import google.generativeai as genai
11
  class QAInfer:
12
  def __init__(self):
13
  torch.cuda.empty_cache()