Spaces:
				
			
			
	
			
			
		Sleeping
		
	
	
	
			
			
	
	
	
	
		
		
		Sleeping
		
	
		Hugo
		
	commited on
		
		
					Commit 
							
							·
						
						96b13ff
	
1
								Parent(s):
							
							671d64d
								
update model v2 (w gen headlines)
Browse files
    	
        app.py
    CHANGED
    
    | @@ -85,7 +85,8 @@ if st.button("Generate Headline"): | |
| 85 | 
             
                if content.strip():
         | 
| 86 | 
             
                    if not video_summary.strip():
         | 
| 87 | 
             
                        video_summary = ''
         | 
| 88 | 
            -
                    prompt = process_prompt(tokenizer, content, video_summary, guidelines)
         | 
|  | |
| 89 | 
             
                    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
         | 
| 90 |  | 
| 91 | 
             
                    st.write("### Generated 5 Potential Headlines:")
         | 
|  | |
| 85 | 
             
                if content.strip():
         | 
| 86 | 
             
                    if not video_summary.strip():
         | 
| 87 | 
             
                        video_summary = ''
         | 
| 88 | 
            +
                    # prompt = process_prompt(tokenizer, content, video_summary, guidelines)
         | 
| 89 | 
            +
                    prompt = process_prompt(tokenizer, content, video_summary)
         | 
| 90 | 
             
                    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
         | 
| 91 |  | 
| 92 | 
             
                    st.write("### Generated 5 Potential Headlines:")
         |