Hugging Face's logo

Spaces:
nileshhanotia
/
Smol
Runtime error

App Files Community
nileshhanotia commited on
Commit
231782d
·
verified ·
1 Parent(s): a6c9b5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,9 +1,8 @@
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
- # Load the model and tokenizer
5
- tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct")
6
- model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct")
7
 
8
  # Streamlit UI
9
  st.title("ChatGPT Clone")
 
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
+ tokenizer = AutoTokenizer.from_pretrained("AIDC-AI/Marco-o1")
5
+ model = AutoModelForCausalLM.from_pretrained("AIDC-AI/Marco-o1")
 
6
 
7
  # Streamlit UI
8
  st.title("ChatGPT Clone")