Update src/streamlit_app.py
Browse files- 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
|