Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Commit 
							
							·
						
						8a609d3
	
1
								Parent(s):
							
							78edf01
								
i forgor
Browse files
    	
        app.py
    CHANGED
    
    | @@ -18,7 +18,7 @@ processor = AutoProcessor.from_pretrained(MODEL_ID) | |
| 18 | 
             
            def tag_image(image, max_new_tokens=128, temperature=1, top_p=1, min_p=0):
         | 
| 19 | 
             
                inputs = processor(images=image, text="<image>tag en", return_tensors="pt").to("cuda")
         | 
| 20 | 
             
                streamer = TextIteratorStreamer(tokenizer=processor.tokenizer, skip_prompt=True)
         | 
| 21 | 
            -
                generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=max_new_tokens, use_cache=True, cache_implementation="hybrid", temperature=temperature, top_p=top_p, min_p=min_p)
         | 
| 22 | 
             
                thread = Thread(target=model.generate, kwargs=generation_kwargs)
         | 
| 23 |  | 
| 24 | 
             
                text = ""
         | 
|  | |
| 18 | 
             
            def tag_image(image, max_new_tokens=128, temperature=1, top_p=1, min_p=0):
         | 
| 19 | 
             
                inputs = processor(images=image, text="<image>tag en", return_tensors="pt").to("cuda")
         | 
| 20 | 
             
                streamer = TextIteratorStreamer(tokenizer=processor.tokenizer, skip_prompt=True)
         | 
| 21 | 
            +
                generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=max_new_tokens, use_cache=True, cache_implementation="hybrid", do_sample=True, temperature=temperature, top_p=top_p, min_p=min_p)
         | 
| 22 | 
             
                thread = Thread(target=model.generate, kwargs=generation_kwargs)
         | 
| 23 |  | 
| 24 | 
             
                text = ""
         | 
