Ezi Ozoani commited on
Commit
fbdf912
1 Parent(s): 713078a
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -32,7 +32,8 @@ def img_to_bytes(img_path):
32
  # sidebar
33
 
34
  def load_model():
35
- pipeline = pipeline(task="image-classification", model="distilgpt2")
 
36
 
37
  def cs_sidebar():
38
 
@@ -294,6 +295,8 @@ were utilized to estimate the carbon impact.*
294
  # Placeholders, help, and options
295
 
296
  col2.subheader('Placeholders, help, and anything else')
 
 
297
  col2.code('''
298
 
299
  ''')
 
32
  # sidebar
33
 
34
  def load_model():
35
+ model_out = pipeline(task="image-classification", model="distilgpt2")
36
+ return model_out
37
 
38
  def cs_sidebar():
39
 
 
295
  # Placeholders, help, and options
296
 
297
  col2.subheader('Placeholders, help, and anything else')
298
+ pipeline = load_model()
299
+
300
  col2.code('''
301
 
302
  ''')