Spaces:

Navyabhat
/

Capstone_Project

Sleeping

Navyabhat commited on Jan 28

Commit

0efe0ff

•

1 Parent(s): 118f209

Update inference/main.py

Files changed (1) hide show

inference/main.py CHANGED Viewed

@@ -72,7 +72,7 @@ class WhisperWithProjection:
 class MultiModalPhi2:
     def __init__(
         self,
-        modelname_or_path="RaviNaik/Llava-Phi2",
         temperature=0.2,
         max_new_tokens=1024,
         device="cuda:0",

 class MultiModalPhi2:
     def __init__(
         self,
+        modelname_or_path="Navyabhat/Llava-Phi2",
         temperature=0.2,
         max_new_tokens=1024,
         device="cuda:0",