Update app.py

app.py

@@ -83,7 +83,7 @@ def load_summarizer():
     return summarizer
 
 # generator, tokenizer = load_generator()
-generator = pipeline("text-generation", model="microsoft/BioGPT")
+generator = pipeline("text-generation", model="mistralai/Mistral-7B-v0.1")
 summarizer = load_summarizer()
 
 image_file = st.sidebar.file_uploader("Upload an Image (.jpg, .jpeg, .png)", type=["jpg", "jpeg", "png"])
@@ -167,14 +167,26 @@ with tab2:
         #     # Extract the generated part of the text
         #     generated_response = generated_text[prompt_length:]
         #     st.session_state.chat_response = generated_response
-
         
         # medical_term = "myocardial infarction"
-        medical_term = prompt
-        description = generator(f"Explain the medical term: {medical_term}")
+        # medical_term = prompt
+        # description = generator(f"Explain the medical term: {medical_term}")
+
+        user_input = st.text_area("Enter your prompt:", prompt)
+        # Slider for controlling the length of the generated text
+        max_length = st.slider("Max length of generated text", 50, 500, 100)   
+        # Button to generate text
+        if st.button("Generate Text"):
+            with st.spinner("Generating text..."):
+                # Generate text using the model
+                output = generator(user_input, max_length=max_length, num_return_sequences=1)
+                generated_text = output[0]['generated_text']
+                # Display the generated text
+                st.subheader("Generated Text:")
+                st.write(generated_text)
         
         # st.session_state.chat_response = outputs
-        st.session_state.chat_response = description
+        st.session_state.chat_response = generated_text
         # Display the generated description
         st.subheader("Generated Description:")
         st.write(prompt, '--')