Sacof commited on
Commit
cc9256b
·
verified ·
1 Parent(s): 1909fbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def is_medical_question(message: str) -> bool:
75
  message_lower = message.lower()
76
  return any(keyword in message_lower for keyword in MEDICAL_KEYWORDS)
77
 
78
- @spaces.GPU(duration=120)
79
  def generate_stream(message: str, history: List[Tuple[str, str]]):
80
  _ensure_loaded()
81
  messages = [{"role": "system", "content": MEDICAL_SYSTEM_PROMPT}]
 
75
  message_lower = message.lower()
76
  return any(keyword in message_lower for keyword in MEDICAL_KEYWORDS)
77
 
78
+ @spaces.GPU()
79
  def generate_stream(message: str, history: List[Tuple[str, str]]):
80
  _ensure_loaded()
81
  messages = [{"role": "system", "content": MEDICAL_SYSTEM_PROMPT}]