Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -74,12 +74,6 @@ def load_model(model_type):
|
|
74 |
except Exception as e:
|
75 |
st.error(f"Error loading model: {str(e)}")
|
76 |
raise
|
77 |
-
|
78 |
-
model.eval()
|
79 |
-
return model, tokenizer
|
80 |
-
except Exception as e:
|
81 |
-
st.error(f"Error loading model: {str(e)}")
|
82 |
-
raise
|
83 |
|
84 |
def cleanup_model(model, tokenizer):
|
85 |
"""Properly cleanup model resources"""
|
|
|
74 |
except Exception as e:
|
75 |
st.error(f"Error loading model: {str(e)}")
|
76 |
raise
|
|
|
|
|
|
|
|
|
|
|
|
|
77 |
|
78 |
def cleanup_model(model, tokenizer):
|
79 |
"""Properly cleanup model resources"""
|