Spaces:
Sleeping
Sleeping
AjithBharadwaj
commited on
Commit
•
97270d3
1
Parent(s):
2f917dc
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def main():
|
|
43 |
word_count = st.sidebar.slider("Number of Words", min_value=50, max_value=1000, value=200, step=50)
|
44 |
|
45 |
if st.sidebar.button("Generate Blog"):
|
46 |
-
model_id = "
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_id,token =HF_TOKEN )
|
48 |
model = AutoModelForCausalLM.from_pretrained(model_id,token =HF_TOKEN )
|
49 |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,max_new_tokens=1000)
|
|
|
43 |
word_count = st.sidebar.slider("Number of Words", min_value=50, max_value=1000, value=200, step=50)
|
44 |
|
45 |
if st.sidebar.button("Generate Blog"):
|
46 |
+
model_id = "google/gemma-1.1-7b-it"
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_id,token =HF_TOKEN )
|
48 |
model = AutoModelForCausalLM.from_pretrained(model_id,token =HF_TOKEN )
|
49 |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,max_new_tokens=1000)
|