Spaces:
Running
Running
Rename app to app.py
Browse files- app → app.py +1 -1
app → app.py
RENAMED
@@ -1,7 +1,7 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
#
|
5 |
@st.cache_resource()
|
6 |
def load_model():
|
7 |
return pipeline("text2text-generation", model="Salesforce/codet5p-220m")
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
#to load the model pipeline
|
5 |
@st.cache_resource()
|
6 |
def load_model():
|
7 |
return pipeline("text2text-generation", model="Salesforce/codet5p-220m")
|