Update app.py

app.py

@@ -7,11 +7,8 @@ IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
 IS_SPACE = os.environ.get("SPACE_ID", None) is not None
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-#dtype = torch.float16
 LOW_MEMORY = os.getenv("LOW_MEMORY", "0") == "1"
 print(f"Using device: {device}")
-#print(f"Using dtype: {dtype}")
 print(f"low memory: {LOW_MEMORY}")
 model_name = "ruslanmv/Medical-Llama3-8B"
 # Move model and tokenizer to the CUDA device
@@ -35,8 +32,6 @@ def askme(symptoms, question):
     assistant_response =response_text.split("assistant")[1].strip().split("user")[0].strip()   
     return assistant_response
 
-
-
 # Example usage
 symptoms = '''\
 I'm a 35-year-old male and for the past few months, I've been experiencing fatigue,
@@ -46,45 +41,9 @@ question = '''\
 Could these symptoms be related to hypothyroidism?
 If so, what steps should I take to get a proper diagnosis and discuss treatment options?
 '''
-
-
 examples = [
     [symptoms, question]
 ]
-'''
-iface = gr.Interface(
-    fn=askme,
-    inputs=["text", "text"],
-    outputs="text",
-    examples=examples,
-    title="Medical AI Chatbot",
-    description="Ask me a medical question!"
-)
-
-iface.launch()
-'''
-
-'''
-
-iface = gr.Interface(
-    fn=askme,
-    inputs=[
-        gr.Textbox(label="Symptoms", placeholder="Enter symptoms here"),
-        gr.Textbox(label="Question", placeholder="Enter question here")
-    ],
-    outputs="text",
-    examples=examples,
-    title="Medical AI Chatbot",
-    description="Ask me a medical question!",
-    css=css
-)
-
-iface.launch()
-'''
-
-
-
-import gradio as gr
 
 css = """
 /* General Container Styles */
@@ -147,3 +106,36 @@ with gr.Blocks(css=css) as interface:
 
 interface.launch()
 
+
+
+
+'''
+iface = gr.Interface(
+    fn=askme,
+    inputs=["text", "text"],
+    outputs="text",
+    examples=examples,
+    title="Medical AI Chatbot",
+    description="Ask me a medical question!"
+)
+
+iface.launch()
+'''
+
+'''
+
+iface = gr.Interface(
+    fn=askme,
+    inputs=[
+        gr.Textbox(label="Symptoms", placeholder="Enter symptoms here"),
+        gr.Textbox(label="Question", placeholder="Enter question here")
+    ],
+    outputs="text",
+    examples=examples,
+    title="Medical AI Chatbot",
+    description="Ask me a medical question!",
+    css=css
+)
+
+iface.launch()
+'''