rashmi commited on
Commit
4b09f99
·
1 Parent(s): 294768e
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -41,8 +41,7 @@ theme = gr.themes.Monochrome(
41
  font=[gr.themes.GoogleFont("Open Sans"), "ui-sans-serif", "system-ui", "sans-serif"],
42
  )
43
 
44
- def do_inference(full_text):
45
- ### Load the model
46
  class CFG:
47
  num_workers = os.cpu_count()
48
  llm_backbone = "HuggingFaceH4/zephyr-7b-beta"
@@ -154,11 +153,14 @@ def do_inference(full_text):
154
  return output.logits
155
 
156
  model = CustomModel()
 
 
 
157
 
158
 
159
  return "result"
160
 
161
- ### End Load the model
162
 
163
 
164
 
 
41
  font=[gr.themes.GoogleFont("Open Sans"), "ui-sans-serif", "system-ui", "sans-serif"],
42
  )
43
 
44
+ ### Load the model
 
45
  class CFG:
46
  num_workers = os.cpu_count()
47
  llm_backbone = "HuggingFaceH4/zephyr-7b-beta"
 
153
  return output.logits
154
 
155
  model = CustomModel()
156
+ ### End Load the model
157
+
158
+ def do_inference(full_text):
159
 
160
 
161
  return "result"
162
 
163
+
164
 
165
 
166