vilarin commited on
Commit
9922e66
·
verified ·
1 Parent(s): 4089b9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -11,7 +11,6 @@ import base64
11
 
12
  # Define the model to be used
13
 
14
-
15
  DESCRIPTION = '''
16
  <div>
17
  <h1 style="text-align: center;">Medster - Medical Diagnostic Assistant</h1>
@@ -33,6 +32,13 @@ footer {
33
 
34
  LICENSE = '[Medster](https://huggingface.co/spaces/vilarin/Medster)'
35
 
 
 
 
 
 
 
 
36
  def endpoints(api_key):
37
  if api_key is not None:
38
  if api_key.startswith('sk-'):
@@ -44,11 +50,6 @@ def endpoints(api_key):
44
  endpoint = 'GOOGLE'
45
  return model_name, endpoint
46
 
47
- def read(filename):
48
- with open(filename) as f:
49
- data = f.read()
50
- return data
51
-
52
  def process_text(api_key, text_input, unit):
53
  model_name, endpoint = endpoints(api_key)
54
  if text_input and endpoint == 'OPENAI':
@@ -107,7 +108,6 @@ def process_image(api_key, image_input, unit):
107
  return response.text
108
 
109
  def main(unit="", api_key="", text_input="", image_input=None):
110
- SYS_PROMPT = read('system_prompt.txt')
111
  if text_input and image_input is None:
112
  return process_text(api_key, text_input, unit)
113
  elif image_input is not None:
 
11
 
12
  # Define the model to be used
13
 
 
14
  DESCRIPTION = '''
15
  <div>
16
  <h1 style="text-align: center;">Medster - Medical Diagnostic Assistant</h1>
 
32
 
33
  LICENSE = '[Medster](https://huggingface.co/spaces/vilarin/Medster)'
34
 
35
+ def read(filename):
36
+ with open(filename) as f:
37
+ data = f.read()
38
+ return data
39
+
40
+ SYS_PROMPT = read('system_prompt.txt')
41
+
42
  def endpoints(api_key):
43
  if api_key is not None:
44
  if api_key.startswith('sk-'):
 
50
  endpoint = 'GOOGLE'
51
  return model_name, endpoint
52
 
 
 
 
 
 
53
  def process_text(api_key, text_input, unit):
54
  model_name, endpoint = endpoints(api_key)
55
  if text_input and endpoint == 'OPENAI':
 
108
  return response.text
109
 
110
  def main(unit="", api_key="", text_input="", image_input=None):
 
111
  if text_input and image_input is None:
112
  return process_text(api_key, text_input, unit)
113
  elif image_input is not None: