..

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from pathlib import Path
 import base64
+#import transformers 
 from transformers import pipeline
 from PIL import Image
 
@@ -17,6 +18,7 @@ st.set_page_config(
 def main():
     cs_sidebar()
     cs_body()
+    load_model()
 
     return None
 
@@ -29,6 +31,9 @@ def img_to_bytes(img_path):
 
 # sidebar
 
+def load_model():
+    pipeline = pipeline(task="image-classification", model="distilgpt2")
+
 def cs_sidebar():
 
    
@@ -284,7 +289,7 @@ were utilized to estimate the carbon impact.*
 
 
 
-    pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+    #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
     # Placeholders, help, and options
 

requirements.txt

@@ -1,2 +1,3 @@
 transformers
-torch
+torch
+transformers-interpret