Praveen0309 commited on
Commit
245d0d6
·
1 Parent(s): 2afa2ba

Application1.2

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -1,15 +1,10 @@
1
  import torch
2
- from transformers import AutoTokenizer, AutoProcessor, TrainingArguments, LlavaForConditionalGeneration, BitsAndBytesConfig
3
- from trl import SFTTrainer
4
- from peft import LoraConfig, PeftModel
5
- from PIL import Image
6
- import requests
7
  from deep_translator import GoogleTranslator
8
  import gradio as gr
9
- import PIL.Image
10
  import base64
11
- import time
12
- import os
13
 
14
 
15
  model_id = "HuggingFaceH4/vsft-llava-1.5-7b-hf-trl"
 
1
  import torch
2
+ from transformers import AutoProcessor, LlavaForConditionalGeneration, BitsAndBytesConfig
3
+ from peft import PeftModel
 
 
 
4
  from deep_translator import GoogleTranslator
5
  import gradio as gr
 
6
  import base64
7
+
 
8
 
9
 
10
  model_id = "HuggingFaceH4/vsft-llava-1.5-7b-hf-trl"