Update app.py

app.py

@@ -1,33 +1,19 @@
 
 import streamlit as st
 
-streamlit run medicine_info_app.py
-
-from transformers import pipeline
-
-# Load the pre-trained model for text generation
-nlp = pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
-
 # Define information about medicines (replace with real data)
 medicine_info = {
-    "topic": {
-        "composition": "topic primarily contains composition as its active ingredient.",
-        "uses": "topic is commonly used to cure which disease.",
-        "purpose": "topic is primarily used to cure various conditions."
+    "aspirin": {
+        "composition": "Aspirin primarily contains acetylsalicylic acid as its active ingredient.",
+        "uses": "Aspirin is commonly used to reduce pain, inflammation, and fever.",
+        "purpose": "Aspirin is primarily used to treat headaches, muscle pain, and arthritis."
     },
-    
     # Add more medicines and their information as needed
 }
 
 def get_medicine_info(medicine_name):
     return medicine_info.get(medicine_name.lower())
 
-def generate_medicine_info(medicine_name):
-    # For simplicity, we'll generate a response using a pre-trained language model
-    prompt = f"Provide information about {medicine_name} medicine."
-    response = nlp(prompt, max_length=100)[0]['generated_text']
-    return response
-
 def main():
     st.title("Medicine Information Finder")
 
@@ -47,13 +33,10 @@ def main():
                 st.subheader("Purpose:")
                 st.write(medicine_data["purpose"])
             else:
-                # Generate information using a pre-trained language model
                 st.error("Sorry, information about this medicine is not available.")
-                st.info("Generating information using a pre-trained model...")
-                generated_info = generate_medicine_info(medicine_name)
-                st.write(generated_info)
         elif submitted and not medicine_name:
             st.error("Please provide the name of the medicine.")
 
 if __name__ == "__main__":
     main()
+