ayush200399391001 commited on
Commit
8978dc5
·
verified ·
1 Parent(s): 52637ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -5,6 +5,10 @@ from datetime import datetime
5
  from transformers import pipeline
6
  import pandas as pd
7
  import re
 
 
 
 
8
 
9
  # Set page config FIRST
10
  st.set_page_config(page_title="AI Medical Consultancy", layout="wide")
 
5
  from transformers import pipeline
6
  import pandas as pd
7
  import re
8
+ import torch
9
+
10
+ # Force CPU usage
11
+ device = torch.device("cpu")
12
 
13
  # Set page config FIRST
14
  st.set_page_config(page_title="AI Medical Consultancy", layout="wide")