Spaces:

CallmeKaito
/

BrainrotGPT

Running

CallmeKaito commited on about 1 month ago

Commit

c3272a9

verified ·

1 Parent(s): d82202f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 # Load model and tokenizer once using caching
 @st.cache_resource

 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
+import torch
 # Load model and tokenizer once using caching
 @st.cache_resource