Update app.py

app.py

@@ -12,6 +12,7 @@ from indic_transliteration import sanscript
 from indic_transliteration.sanscript import transliterate
 import openai
 import torch
+import warnings
 
 # Detect if GPU is available
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -44,7 +45,9 @@ if text_generation_tokenizer.pad_token is None:
 API_URL = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
 
 # Load the trained sentiment analysis model and preprocessing steps
-model = joblib.load('model.pkl')
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    model = joblib.load('model.pkl')
 
 # Function to query Hugging Face API
 def query(payload, max_retries=5):