123

app.py

@@ -1,7 +1,24 @@
 import gradio as gr
+import torch.nn as nn
+from huggingface_hub import PyTorchModelHubMixin
+
+# os.system('pip install torch')
 
 def greet(name):
     return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+class MyModel(nn.Module, PyTorchModelHubMixin):
+    def __init__(self):
+        super(MyModel, self).__init__()
+        self.linear = nn.Linear(20,50)
+
+    def forward(self, x):
+        return x+2
+        
+model = MyModel.from_pretrained('WeixuanYuan/test_upload_private', use_auth_token='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')
+
+def plus_two(input):
+    return model(int(input))
+
+iface = gr.Interface(fn=plus_two, inputs="text", outputs="text")
 iface.launch()