Update app.py

app.py

@@ -65,26 +65,31 @@ class Model:
       generated = self.model.generate(tokens, num_beams=4, max_length=256)
       tgt_text = self.tokenizer.decode(generated[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
       summary = str(tgt_text).strip('[]""')
+      
       if 'barchart' in summary:
         summary.replace('barchart','statistic')
-      if 'bar graph' in summary:
+      elif 'bar graph' in summary:
         summary.replace('bar graph','statistic')
-      if 'bar plot' in summary:
+      elif 'bar plot' in summary:
         summary.replace('bar plot','statistic')
-      if 'scatter plot' in summary:
+      elif 'scatter plot' in summary:
         summary.replace('scatter plot','statistic')
-      if 'scatter graph' in summary:
+      elif 'scatter graph' in summary:
         summary.replace('scatter graph','statistic')
-      if 'scatterchart' in summary:
+      elif 'scatterchart' in summary:
         summary.replace('scatter chart','statistic')
-      if 'line plot' in summary:
+      elif 'line plot' in summary:
         summary.replace('line plot','statistic')
-      if 'line graph' in summary:
+      elif 'line graph' in summary:
         summary.replace('line graph','statistic')
-      if 'linechart' in summary:
+      elif 'linechart' in summary:
         summary.replace('linechart','statistic')
+      
       if 'graph' in summary:
         summary.replace('graph','statistic')
+        
+        
+        
       return summary
       
 st.title('Chart and Data Summarization')