Update app.py

app.py

@@ -6,7 +6,6 @@ warnings.simplefilter('ignore')
 logging.disable(logging.WARNING)
 
 
-
 def predict(image):
     cap = pipeline('image-to-text')
     caption = cap(image)
@@ -16,15 +15,22 @@ def predict(image):
       pipe = pipeline('text-classification')
       senti = pipe(shortened)
       return senti
+
     caption_string = str(caption)
-    shortened = caption_string[21:-3]
-    formatted_capt = caption_string.replace("[{'generated_text': '" , "The image is of, ")
-    formatted_capt = formatted_capt.replace("'}]" , "")
-    sentiment = sentiment_analysis(shortened)
+
+    sentiment = sentiment_analysis(caption_string[21:-1])
+
+
+
     sentiment_string = ''.join(str(e) for e in sentiment)
-    formated_senti = sentiment_string.replace("{'label': ", ", The Tone of the image(sentiment) after analysisng the caption is that is it is ")
-    formated_senti = formated_senti.replace("'score': ", "in nature with an average percentage of ")
-    output = formatted_capt + formated_senti[0:-2] + '%'
+
+    formatted_cap = caption_string.replace("[{'generated_text': '" , "the image is of, ")
+
+    formated_senti = sentiment_string.replace("{'label': ", ", The Tone of the image is mostly ")
+
+    formated_senti = formated_senti.replace("score': ", " with approx : ")
+
+    output = formatted_cap + formated_senti[1:-15] + '% score'
     return output