Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,11 +34,11 @@ except Exception as e:
|
|
| 34 |
app = Flask(__name__)
|
| 35 |
logging.info("Flask app initialized")
|
| 36 |
|
| 37 |
-
# Salesforce credentials
|
| 38 |
-
SF_USERNAME =
|
| 39 |
-
SF_PASSWORD =
|
| 40 |
-
SF_SECURITY_TOKEN =
|
| 41 |
-
SF_INSTANCE_URL =
|
| 42 |
|
| 43 |
# Global variable for Salesforce connection
|
| 44 |
sf = None
|
|
@@ -81,11 +81,11 @@ def connect_to_salesforce():
|
|
| 81 |
def load_huggingface_model():
|
| 82 |
global summarizer
|
| 83 |
if summarizer is None:
|
| 84 |
-
logging.info("Loading Hugging Face model...")
|
| 85 |
start_time = time.time()
|
| 86 |
try:
|
| 87 |
device = 0 if torch.cuda.is_available() else -1
|
| 88 |
-
summarizer = pipeline("summarization", model="t5-small", device=device)
|
| 89 |
logging.info(f"Hugging Face model loaded successfully in {time.time() - start_time:.2f} seconds on device: {'GPU' if device == 0 else 'CPU'}")
|
| 90 |
except Exception as e:
|
| 91 |
logging.error(f"Failed to load Hugging Face model: {str(e)}")
|
|
|
|
| 34 |
app = Flask(__name__)
|
| 35 |
logging.info("Flask app initialized")
|
| 36 |
|
| 37 |
+
# Salesforce credentials (hardcoded for testing)
|
| 38 |
+
SF_USERNAME = "multi-devicelabopsdashboard@sathkrutha.com"
|
| 39 |
+
SF_PASSWORD = "Team@1234"
|
| 40 |
+
SF_SECURITY_TOKEN = "BXgWWNXjvc3zJmVv2O7JfBqCc"
|
| 41 |
+
SF_INSTANCE_URL = "https://multi-devicelabopsdashboard-dev-ed.develop.my.salesforce.com"
|
| 42 |
|
| 43 |
# Global variable for Salesforce connection
|
| 44 |
sf = None
|
|
|
|
| 81 |
def load_huggingface_model():
|
| 82 |
global summarizer
|
| 83 |
if summarizer is None:
|
| 84 |
+
logging.info("Loading Hugging Face model (t5-small)...")
|
| 85 |
start_time = time.time()
|
| 86 |
try:
|
| 87 |
device = 0 if torch.cuda.is_available() else -1
|
| 88 |
+
summarizer = pipeline("summarization", model="t5-small", device=device)
|
| 89 |
logging.info(f"Hugging Face model loaded successfully in {time.time() - start_time:.2f} seconds on device: {'GPU' if device == 0 else 'CPU'}")
|
| 90 |
except Exception as e:
|
| 91 |
logging.error(f"Failed to load Hugging Face model: {str(e)}")
|