pittman commited on
Commit
12d1e81
·
verified ·
1 Parent(s): 908bed4

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -83,7 +83,7 @@ seminalchurch@gmail.com
83
 
84
  # Load local model for inference (free, no API)
85
  try:
86
- generator = pipeline("text-generation", model="gpt2")
87
  except Exception as e:
88
  st.write(f"Error loading model: {str(e)}. Clearing cache and retrying may help.")
89
  generator = None
 
83
 
84
  # Load local model for inference (free, no API)
85
  try:
86
+ generator = pipeline("text-generation", model="gpt2", device=-1) # Force CPU loading to avoid meta tensor issues
87
  except Exception as e:
88
  st.write(f"Error loading model: {str(e)}. Clearing cache and retrying may help.")
89
  generator = None