Update app.py

app.py

@@ -8,7 +8,6 @@ import gradio as gr
 from utils import load_json, init_logger
 from demo import ConversationalAgent, CustomTheme
 
-
 os.system('git lfs install')
 os.system("git clone https://huggingface.co/dstars/InternVL_Food")
 
@@ -34,8 +33,11 @@ def main():
     # food examples
     food_examples = load_json(FOOD_EXAMPLES)
     
+    # Set the device to CPU explicitly
+    device = 'cpu'
     agent = ConversationalAgent(model_path=MODEL_PATH,
-                                outputs_dir=OUTPUT_PATH)
+                                outputs_dir=OUTPUT_PATH,
+                                device=device)  # Pass the device to the agent
     
     theme = CustomTheme()
     
@@ -48,7 +50,6 @@ def main():
     with gr.Blocks(theme) as demo_chatbot:
         for title in titles:
             gr.Markdown(title)
-        # gr.Markdown(article)
         gr.Markdown(language)
         
         with gr.Row():
@@ -119,4 +120,4 @@ def main():
     
 
 if __name__ == "__main__":
-    main()
+    main()