jacksonwambali commited on
Commit
693b41f
·
verified ·
1 Parent(s): fadbeaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,9 +5,11 @@ import folium
5
  from fastai.vision.all import *
6
  from groq import Groq
7
  from PIL import Image
 
 
8
 
9
  # Load the trained model
10
- learn = load_learner('export.pkl', safe=True)
11
  labels = learn.dls.vocab
12
 
13
  # Initialize Groq client
 
5
  from fastai.vision.all import *
6
  from groq import Groq
7
  from PIL import Image
8
+ from fasttransform import Pipeline
9
+
10
 
11
  # Load the trained model
12
+ learn = load_learner('export.pkl', safe=True) # Optionally set safe=True for more security
13
  labels = learn.dls.vocab
14
 
15
  # Initialize Groq client