RathodHarish commited on
Commit
a6bde15
·
verified ·
1 Parent(s): ad87f6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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 = os.getenv('SF_USERNAME', 'your_salesforce_username')
39
- SF_PASSWORD = os.getenv('SF_PASSWORD', 'your_salesforce_password')
40
- SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN', 'your_security_token')
41
- SF_INSTANCE_URL = os.getenv('SF_INSTANCE_URL', 'https://login.salesforce.com')
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) # Use a smaller model
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)}")