Update app.py

app.py

@@ -16,8 +16,6 @@ def get_model():
 feature_extractor = AutoFeatureExtractor.from_pretrained(CHECKPOINT)
 tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)
 
-model.eval()
-
 st.title("Image Captioning with ViT & GePpeTto 🇮🇹")
 
 st.sidebar.markdown("## Generation parameters")
@@ -49,6 +47,7 @@ def generate_caption(url):
     image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
     inputs = feature_extractor(image, return_tensors="pt")
     model = get_model()
+    model.eval()
     generated_ids = model.generate(
         inputs["pixel_values"],
         max_length=20,