Ritesh-hf commited on
Commit
4968746
·
verified ·
1 Parent(s): 94c6109

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -393,7 +393,7 @@ def answer_formatter_node(question, context):
393
  CURR_CONTEXT = ''
394
  CURR_SESSION_KEY = generate_session_key()
395
 
396
- @spaces.GPU
397
  def get_answer(image=[], message='', sessionID='abc123'):
398
  global CURR_CONTEXT
399
  global CURR_SESSION_KEY
@@ -542,7 +542,7 @@ def handle_message(data):
542
 
543
 
544
 
545
- @spaces.GPU
546
  def respond_to_user(image, message):
547
  # Process the image and message here
548
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
393
  CURR_CONTEXT = ''
394
  CURR_SESSION_KEY = generate_session_key()
395
 
396
+ # @spaces.GPU
397
  def get_answer(image=[], message='', sessionID='abc123'):
398
  global CURR_CONTEXT
399
  global CURR_SESSION_KEY
 
542
 
543
 
544
 
545
+ # @spaces.GPU
546
  def respond_to_user(image, message):
547
  # Process the image and message here
548
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")