Spaces:

Ritesh-hf
/

Nutrigenics-chatbot

Sleeping

OmkarThawakar commited on Oct 12, 2024

Commit

3cf14d0

1 Parent(s): c2d589f

change device setting

Files changed (1) hide show

app.py CHANGED Viewed

@@ -182,8 +182,7 @@ custom_css = """
 def respond_to_user(image, message):
     # Process the image and message here
-    # For demonstration, I'll just return a simple text response
-    chat = Chat(model,transform,df,tar_img_feats)
     chat.encode_image(image)
     data = chat.ask()
     formated_input = {

 def respond_to_user(image, message):
     # Process the image and message here
+    chat = Chat(model,transform,df,tar_img_feats, device)
     chat.encode_image(image)
     data = chat.ask()
     formated_input = {