prithivMLmods commited on
Commit
c93b234
·
verified ·
1 Parent(s): 00dda18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -32,7 +32,6 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
32
 
33
  # Load text-only model and tokenizer (Pocket Llama)
34
  model_id = "prithivMLmods/Pocket-Llama2-3.2-3B-Instruct"
35
- #model_id = "prithivMLmods/Flerovium-2B"
36
  tokenizer = AutoTokenizer.from_pretrained(model_id)
37
  model = AutoModelForCausalLM.from_pretrained(
38
  model_id,
@@ -286,12 +285,7 @@ demo = gr.ChatInterface(
286
  ["Write the code that converts temperatures between Celsius and Fahrenheit in short"],
287
  [{"text": "Create a short story based on the image.", "files": ["examples/1.jpg"]}],
288
  ["@JennyNeural Who was Nikola Tesla and what were his contributions?"],
289
- [{"text": "@video-infer Describe the video", "files": ["examples/Missing.mp4"]}],
290
- # [{"text": "@video-infer Describe the Ad", "files": ["examples/coca.mp4"]}],
291
- # ["@GuyNeural Explain how rainbows are formed."],
292
- # ["@PalomaNeural What is the water cycle?"],
293
- # ["@AlonsoNeural Who was Pablo Picasso and why is he famous?"],
294
- # ["@MadhurNeural What are the key principles of Ayurveda?"]
295
  ],
296
  cache_examples=False,
297
  description="# **Callisto OCR**",
 
32
 
33
  # Load text-only model and tokenizer (Pocket Llama)
34
  model_id = "prithivMLmods/Pocket-Llama2-3.2-3B-Instruct"
 
35
  tokenizer = AutoTokenizer.from_pretrained(model_id)
36
  model = AutoModelForCausalLM.from_pretrained(
37
  model_id,
 
285
  ["Write the code that converts temperatures between Celsius and Fahrenheit in short"],
286
  [{"text": "Create a short story based on the image.", "files": ["examples/1.jpg"]}],
287
  ["@JennyNeural Who was Nikola Tesla and what were his contributions?"],
288
+ [{"text": "@video-infer Describe the video", "files": ["examples/Missing.mp4"]}]
 
 
 
 
 
289
  ],
290
  cache_examples=False,
291
  description="# **Callisto OCR**",