pendar02 commited on
Commit
2ecd393
·
verified ·
1 Parent(s): 8f96b1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -6
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"""