Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -53,12 +53,12 @@ def generate(model,tokenizer,instruction,context,streamer):
|
|
53 |
@st.cache_resource
|
54 |
def get_model():
|
55 |
# Create a model of the specified type.
|
56 |
-
return AutoModelForCausalLM.from_pretrained("lagy/
|
57 |
|
58 |
@st.cache_resource
|
59 |
def get_tokenizer():
|
60 |
# Create a model of the specified type.
|
61 |
-
return AutoTokenizer.from_pretrained("lagy/
|
62 |
|
63 |
st.write(" ")
|
64 |
|
|
|
53 |
@st.cache_resource
|
54 |
def get_model():
|
55 |
# Create a model of the specified type.
|
56 |
+
return AutoModelForCausalLM.from_pretrained("lagy/carballo-1.3b-finetuned")
|
57 |
|
58 |
@st.cache_resource
|
59 |
def get_tokenizer():
|
60 |
# Create a model of the specified type.
|
61 |
+
return AutoTokenizer.from_pretrained("lagy/carballo-1.3b-finetuned")
|
62 |
|
63 |
st.write(" ")
|
64 |
|