Update app.py

app.py

@@ -10,7 +10,7 @@ def generate(tokenizer, model, text, features):
     generated = tokenizer("<|startoftext|> <|titlestart|>{}<|titleend|>".format(text), return_tensors="pt").input_ids
     sample_outputs = model.generate(
         generated, do_sample=True, top_k=50, 
-        max_length=300, top_p=features['top_p'], temperature=features['t'] / 100.0, num_return_sequences=features['num'],
+        max_length=features['max_length'], top_p=features['top_p'], temperature=features['t'] / 100.0, num_return_sequences=features['num'],
     )
     for i, sample_output in enumerate(sample_outputs):
         decoded = tokenizer.decode(sample_output, skip_special_tokens=True).split(text)[1]
@@ -39,6 +39,7 @@ def main():
         "num": st.sidebar.slider("Количество комментариев", 0, 20, 1, 1),
         "t": st.sidebar.slider("Температура", 0, 300, 180, 1),
         "top_p": st.sidebar.slider("Top-p", 0, 100, 95, 5),
+        "max_length": st.sidebar.slider("Максимальная длина комментария", 0, 300, 100, 5),
     }
 
     st.sidebar.title("Note")
@@ -56,18 +57,19 @@ def main():
     with st.form(key='my_form'):
         url = st.text_input('Введите url видео на YouTube')
         st.form_submit_button('Готово!')
-
-    params = {"format": "json", "url": url}
-    base_url = "https://www.youtube.com/oembed"
-    query_string = urllib.parse.urlencode(params)
-    base_url = base_url + "?" + query_string
-
-    with urllib.request.urlopen(base_url) as response:
-        response_text = response.read()
-        data = json.loads(response_text.decode())
-        st.write('Video Title: ' + data['title'])
-    st.video(url)
-    generate(tokenizer, model, data['title'], features)
-
+        
+    if url:
+        params = {"format": "json", "url": url}
+        base_url = "https://www.youtube.com/oembed"
+        query_string = urllib.parse.urlencode(params)
+        base_url = base_url + "?" + query_string
+    
+        with urllib.request.urlopen(base_url) as response:
+            response_text = response.read()
+            data = json.loads(response_text.decode())
+            st.write('Video Title: ' + data['title'])
+        st.video(url)
+        generate(tokenizer, model, data['title'], features)
+    
 if __name__ == "__main__":
     main()