Spaces:

Ritesh-hf
/

nutrigenics-flask-chatbot-server

Running on CPU Upgrade

Ritesh-hf commited on Oct 28, 2024

Commit

28ff4fe

1 Parent(s): ab12644

fkjhsfk

Files changed (1) hide show

app.py CHANGED Viewed

@@ -451,7 +451,7 @@ CURR_CONTEXT = ''
 # @spaces.GPU
 def get_answer(image=[], message='', sessionID='abc123'):
     global CURR_CONTEXT
-    if image:
         try:
             # Process the image and message here
             device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 # @spaces.GPU
 def get_answer(image=[], message='', sessionID='abc123'):
     global CURR_CONTEXT
+    if len(image) > 0:
         try:
             # Process the image and message here
             device = torch.device("cuda" if torch.cuda.is_available() else "cpu")