Spaces:
Sleeping
Sleeping
OmPrakashSingh1704
commited on
Commit
•
dd6553b
1
Parent(s):
58d8835
Update app.py
Browse files
app.py
CHANGED
@@ -9,12 +9,12 @@ login(token=os.getenv("TOKEN"))
|
|
9 |
|
10 |
# Load model and tokenizer
|
11 |
model = AutoModelForCausalLM.from_pretrained(
|
12 |
-
"google/gemma-
|
13 |
torch_dtype="auto",
|
14 |
device_map="auto",
|
15 |
)
|
16 |
|
17 |
-
tokenizer = AutoTokenizer.from_pretrained("google/gemma-
|
18 |
if 'recipe' not in st.session_state:
|
19 |
st.session_state.recipe = None
|
20 |
|
|
|
9 |
|
10 |
# Load model and tokenizer
|
11 |
model = AutoModelForCausalLM.from_pretrained(
|
12 |
+
"google/gemma-2b",
|
13 |
torch_dtype="auto",
|
14 |
device_map="auto",
|
15 |
)
|
16 |
|
17 |
+
tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
|
18 |
if 'recipe' not in st.session_state:
|
19 |
st.session_state.recipe = None
|
20 |
|