rajeshlion commited on
Commit
af63dec
·
verified ·
1 Parent(s): 428d7b8

Update app_gpu2.py

Browse files
Files changed (1) hide show
  1. app_gpu2.py +2 -0
app_gpu2.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import openai
3
  from dotenv import load_dotenv
@@ -59,6 +60,7 @@ class cbfs:
59
  return_generated_question=True,
60
  )
61
 
 
62
  def convchain(self, query):
63
  if not query:
64
  return [("User", ""), ("ChatBot", "")]
 
1
+ import spaces
2
  import os
3
  import openai
4
  from dotenv import load_dotenv
 
60
  return_generated_question=True,
61
  )
62
 
63
+ @spaces.GPU
64
  def convchain(self, query):
65
  if not query:
66
  return [("User", ""), ("ChatBot", "")]