Update app.py

app.py

@@ -34,6 +34,7 @@ don't include '.com' in outputs and give different names.
 {}"""
 
 
+
 def print_output(input):
     inputs = tokenizer(
     [
@@ -46,102 +47,130 @@ def print_output(input):
     # outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
     # tokenizer.batch_decode(outputs)
     
-    outputs = model.generate(**inputs, max_new_tokens = 200, use_cache = True, temperature=0.5)
+    outputs = model.generate(**inputs, max_new_tokens = 100, use_cache = True)
+    return tokenizer.batch_decode(outputs)
+
+
+# Streamlit app UI
+st.title("Echo Input Example")
+
+# Text box for user input
+user_input = st.text_input("Type something here:")
+
+# Button to submit input
+if st.button("Submit"):
+    # Reprint the user input
+    st.write(f"You typed: {print_output(user_input)}")
+    
+
+# def print_output(input):
+#     inputs = tokenizer(
+#     [
+#         inf_prompt_1.format(
+#             input, # instruction
+#             "", # output - leave this blank for generation!
+#         ),
+#     ], return_tensors = "pt").to("cuda")
+    
+#     # outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
+#     # tokenizer.batch_decode(outputs)
+    
+#     outputs = model.generate(**inputs, max_new_tokens = 200, use_cache = True, temperature=0.5)
 
-    n1 = tokenizer.batch_decode(outputs)[0].index('### Response:')
-    output_ = tokenizer.batch_decode(outputs)[0][n1+13:].replace('.com','')
-    n2 = output_.index('\n\n')
+#     n1 = tokenizer.batch_decode(outputs)[0].index('### Response:')
+#     output_ = tokenizer.batch_decode(outputs)[0][n1+13:].replace('.com','')
+#     n2 = output_.index('\n\n')
 
-    dom_names = []
+#     dom_names = []
     
-    for name in output_[:n2].split('\n'):
-      if name != '':
-          dom_names.append(name.split('.')[1])
+#     for name in output_[:n2].split('\n'):
+#       if name != '':
+#           dom_names.append(name.split('.')[1])
     
-    s = '\n'.join(dom_names)
-
-    return tokenizer.batch_decode(s)
-
-
-# # Streamlit app UI
-# st.title("Echo Input Example")
-
-# # Text box for user input
-# user_input = st.text_input("Type something here:")
-
-# # Button to submit input
-# if st.button("Submit"):
-#     # Reprint the user input
-#     st.write(f"You typed: {print_output(user_input)}")
-
-
-def main():
-
-    css_dark_mode = """
-    <style>
-        body {
-            background-color: #121212;
-            color: #f8f9fa;
-            font-family: Arial, sans-serif;
-        }
-        .header {
-            background: linear-gradient(90deg, #005c97, #363795);
-            padding: 15px;
-            margin-bottom: 15px;
-            border-radius: 10px;
-            box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);
-            text-align: center;
-        }
-        .header h1 {
-            color: #ffffff;
-            font-size: 36px;
-        }
-        .header h2 {
-            color: #b0b0b0;
-            font-size: 22px;
-            font-family: Georgia, serif;
-        }
-        .stButton > button {
-            background-color: #20c997;
-            color: #ffffff;
-            border: none;
-            border-radius: 8px;
-            padding: 10px 20px;
-            font-size: 16px;
-            cursor: pointer;
-            transition: background-color 0.3s;
-        }
-        .stButton > button:hover {
-            background-color: #17a589;
-        }
-        .success-box {
-            background-color: #A1D6CB;
-            padding: 10px;
-            margin: 10px 0;
-            border-radius: 8px;
-            box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);
-        }
-    </style>
-    """
-    # Inject the custom CSS
-    st.markdown(css_dark_mode, unsafe_allow_html=True)
-
-    # Header Section
-    html_temp = """
-    <div class="header">
-        <h1>Dominious</h1>
-        <h2>AI Based Domain Name Suggestion System</h2>
-    </div>
-    """
-    st.markdown(html_temp, unsafe_allow_html=True)
-    # st.title("Domain Name Suggestion System")
-
-    # Get user input
-    user_input = st.text_input("Describe your business here...")
-
-    # Generate the response
-    if st.button("Generate"):
-        with st.spinner("Generating Domain Names..."):
-            result = print_output(user_input)
-if __name__ == "__main__":
-    main()
+#     s = '\n'.join(dom_names)
+
+#     return tokenizer.batch_decode(s)
+
+
+# # # Streamlit app UI
+# # st.title("Echo Input Example")
+
+# # # Text box for user input
+# # user_input = st.text_input("Type something here:")
+
+# # # Button to submit input
+# # if st.button("Submit"):
+# #     # Reprint the user input
+# #     st.write(f"You typed: {print_output(user_input)}")
+
+
+# def main():
+
+#     css_dark_mode = """
+#     <style>
+#         body {
+#             background-color: #121212;
+#             color: #f8f9fa;
+#             font-family: Arial, sans-serif;
+#         }
+#         .header {
+#             background: linear-gradient(90deg, #005c97, #363795);
+#             padding: 15px;
+#             margin-bottom: 15px;
+#             border-radius: 10px;
+#             box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);
+#             text-align: center;
+#         }
+#         .header h1 {
+#             color: #ffffff;
+#             font-size: 36px;
+#         }
+#         .header h2 {
+#             color: #b0b0b0;
+#             font-size: 22px;
+#             font-family: Georgia, serif;
+#         }
+#         .stButton > button {
+#             background-color: #20c997;
+#             color: #ffffff;
+#             border: none;
+#             border-radius: 8px;
+#             padding: 10px 20px;
+#             font-size: 16px;
+#             cursor: pointer;
+#             transition: background-color 0.3s;
+#         }
+#         .stButton > button:hover {
+#             background-color: #17a589;
+#         }
+#         .success-box {
+#             background-color: #A1D6CB;
+#             padding: 10px;
+#             margin: 10px 0;
+#             border-radius: 8px;
+#             box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);
+#         }
+#     </style>
+#     """
+#     # Inject the custom CSS
+#     st.markdown(css_dark_mode, unsafe_allow_html=True)
+
+#     # Header Section
+#     html_temp = """
+#     <div class="header">
+#         <h1>Dominious</h1>
+#         <h2>AI Based Domain Name Suggestion System</h2>
+#     </div>
+#     """
+#     st.markdown(html_temp, unsafe_allow_html=True)
+#     # st.title("Domain Name Suggestion System")
+
+#     # Get user input
+#     user_input = st.text_input("Describe your business here...")
+
+#     # Generate the response
+#     if st.button("Generate"):
+#         with st.spinner("Generating Domain Names..."):
+#             result = print_output(user_input)
+# if __name__ == "__main__":
+#     main()